题目描述
给定两个由小写字母构成的字符串 A 和 B ,只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果,就返回 true ;否则返回 false 。
示例
输入: A = “ab”, B = “ba”
输出: true
输入: A = “ab”, B = “ab”
输出: false
输入: A = “aa”, B = “aa”
输出: true
输入: A = “aaaaaaabc”, B = “aaaaaaacb”
输出: true
输入: A = “”, B = “aa”
输出: false
解答
根据示例总结的规律:
- 两个字符串长度不等,返回false
- 两个字符串完全相同:
(1)A中无重复字符,返回false
(2)A中有重复字符,可以交换重复字符,返回true - A和B等长,但部分字符不同,不同的字符对数小于等于2,交换一次即可变为相同,返回true
class Solution(object):
def buddyStrings(self, A, B):
"""
:type A: str
:type B: str
:rtype: bool
"""
len_a = len(A)
len_b = len(B)
if len_a != len_b:
return False
diff_a = ""
diff_b = ""
for i in range(0,len_a):
if A[i]!=B[i]:
diff_a = diff_a + A[i]
diff_b = diff_b + B[i]
# A和B相同
if len(diff_a)==0:
# A中有重复元素,则可以交换重复元素
if len(set(A))<len(A):
return True
else:
return False
# A和B中不同的对数超过2
elif len(diff_a)>2:
return False
# A和B中不同的对数小于等于2
else:
# 交换A中元素与B相同
if diff_a[::-1]==diff_b:
return True
else:
return False