给定两个由小写字母构成的字符串 A 和 B ,只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果,就返回 true ;否则返回 false 。
交换字母的定义是取两个下标 i 和 j (下标从 0 开始),只要 i!=j 就交换 A[i] 和 A[j] 处的字符。例如,在 “abcd” 中交换下标 0 和下标 2 的元素可以生成 “cbad” 。
示例 1:
输入: A = "ab", B = "ba"
输出: true
解释: 你可以交换 A[0] = 'a' 和 A[1] = 'b' 生成 "ba",此时 A 和 B 相等。
示例 2:
输入: A = "ab", B = "ab"
输出: false
解释: 你只能交换 A[0] = 'a' 和 A[1] = 'b' 生成 "ba",此时 A 和 B 不相等。
示例 3:
输入: A = "aa", B = "aa"
输出: true
解释: 你可以交换 A[0] = 'a' 和 A[1] = 'a' 生成 "aa",此时 A 和 B 相等。
示例 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, s: str, goal: str) -> bool:
if(len(s) != len(goal)):return False
s,goal,l = list(s),list(goal),[]
if(s == goal):
# 寻找一样的两个
for i in range(1,len(s)):
if(s[i-1] in s[i:]):
return True
return False
else:
for i in range(len(s)):
if(s[i]!=goal[i]):l.append(i)
if(len(l) == 2):break
if(len(l) == 1):return False
# 0 == > 完全一样
if(len(l) == 2):
temp = s[l[0]]
s[l[0]] = s[l[1]]
s[l[1]] = temp
return True if(s == goal) else False
方案二:
官方,思路差不多
class Solution(object):
def buddyStrings(self, A, B):
if len(A) != len(B): return False
if A == B:
seen = set()
for a in A:
if a in seen:
return True
seen.add(a)
return False
else:
pairs = []
for a, b in itertools.izip(A, B):
if a != b:
pairs.append((a, b))
if len(pairs) >= 3: return False
return len(pairs) == 2 and pairs[0] == pairs[1][::-1]