Python字符串替换

给出三行字符串，第一行为被改字符串，第二行为目标字符串，第三行为替换目标的字符串，二三行长度不一定相等。

s1=input()
s2=input()
s3=input()
num=0
for i in s1:
  if i in s2:
    num+=1
    s1=s1.replace(i,s3[num])
    if num==len(s3):
      break
print(s1)

想知道应该怎么修改

s1=input()
s2=input()
s3=input()
s1 = s1.replace(s2,s3)
print(s1)