交换字符串中字符

输入A,B两个字符串，如果交换A中某两个字符后，与B相等，输出1，否则，输出0。

代码：

A=list(input().strip())
B=list(input().strip())
n=len(A)
array=[]
for i in range(n):
    for j in range(i+1,n):
        C=A[:]
        ree=C[i]
        C[i]=C[j]
        C[j]=ree
        array.append(C)
if B in array:
    print(1)
else:
    print(0)