题目原文
给定两个由小写字母构成的字符串 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 仅由小写字母构成。
解题思路
- 若A的长度不等于B的长度,直接return False;
- 当A、B相同时,使用set()判断A中是否有重复的元素,若有直接返回True;
- 当A、B不相同时,使用diff获得不相等的索引列表,通过比较索引判断是否满足条件。
class Solution:
def buddyStrings(self, A: str, B: str) -> bool:
if len(A) != len(B):
return False
if A == B and len(set(A)) < len(A):
return True
A = list(A)
B = list(B)
diff = []
for i in range(len(A)):
if A[i] != B[i]:
diff.append(i)
if len(diff) == 2 and A[diff[0]] == B[diff[1]] and B[diff[0]] == A[diff[1]]:
return True
else:
return False
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/buddy-strings