【题目】
给定两个由小写字母构成的字符串 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 仅由小写字母构成
【代码】
【Python】
执行用时:
44 ms, 在所有 Python3 提交中击败了55.15%的用户
内存消耗:
14.7 MB, 在所有 Python3 提交中击败了65.10%的用户
class Solution:
def buddyStrings(self, A: str, B: str) -> bool:
# 必须交换一次,示例2 ab和ab本身相等,如果非要交换一次就会不相等
if len(A)!=len(B) or len(A)<=1:
return False
count=dict(Counter(A))
d_order=sorted(count.items(),key=lambda x:x[1],reverse=True)
cnt,posA,posB,i=0,-1,-1,0
for x,y in zip(A,B):
if x!=y:
cnt+=1
if cnt==1:
posA=i
elif cnt==2:
posB=i
i+=1
if cnt==0:
return d_order[0][1]>1
return cnt==2 and A[posA]==B[posB] and A[posB]==B[posA]