给定两个由小写字母构成的字符串 A 和 B ,只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果,就返回 true ;否则返回 false 。
示例 1:
输入: A = "ab", B = "ba"
输出: true
示例 2:
输入: A = "ab", B = "ab"
输出: false
示例 3:
输入: A = "aa", B = "aa"
输出: true
示例 4:
输入: A = "aaaaaaabc", B = "aaaaaaacb"
输出: true
示例 5:
输入: A = "", B = "aa"
输出: false
提示:
0 <= A.length <= 20000
0 <= B.length <= 20000
A 和 B 仅由小写字母构成。
class Solution:
def buddyStrings(self, A: str, B: str) -> bool:
if len(A) != len(B):
return 0
index = []
for i in range(len(A)):
if A[i] != B[i]:
index.append(i)
if len(index) == 0:
dicts = {}
for j in A:
if j not in dicts.keys():
dicts[j] = 1
else:
dicts[j] += 1
for item in dicts.values():
if item>1:
return 1
return 0
elif len(index) == 2:
a = A[:index[0]]+A[index[1]]+A[index[0]+1:index[1]]+A[index[0]]+A[index[1]+1:]
if a == B:
return 1
else:
return 0
else:
return 0