给定两个由小写字母构成的字符串 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(object):
def buddyStrings(self, A, B):
"""
:type A: str
:type B: str
:rtype: bool
"""
i = 0
if len(A) == 0 or len(B) == 0 or len(A) != len(B):
return False
n = len(A)
A,B = list(A),list(B)
# different记录AB中不同的字符的索引
different = []
while i<n:
if A[i] != B[i]:
different.append(i)
i+=1
# 如果不同的字符大于两个,直接返回False,因为只能交换一次
if len(different) > 2:
return False
# 如果没有不同字符,只要有一个字符数量大于1,则是亲密字符串
# (因为这个时候可以交换任意字符数量大于等于2的字符)
elif len(different) == 0:
for a in set(A):
if A.count(a)>1:
return True
return False
else:
A[different[0]],A[different[1]] = A[different[1]],A[different[0]]
return A==B