题目描述:
给定两个由小写字母构成的字符串 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/buddy-strings
解题思路及代码:
### 解题思路
根据题意,判断多个条件即可:
1.判断两个字符长度,如不同,直接返回False;
2.一次遍历,找到不同的位置,并保存;
3.当不同的次数为2时,比较两个字符在不同位置下交换是否相等;
4.当不同次数为0时,只需要判断字符中是否有重复的字母出现即可。
### 代码
```python
class Solution(object):
def buddyStrings(self, a, b):
"""
:type a: str
:type b: str
:rtype: bool
"""
if len(a) != len(b):
return False
same_num = []
for i in range(len(a)):
if a[i] != b[i]:
same_num.append(i)
if len(same_num) == 2:
if a[same_num[0]] == b[same_num[1]] and a[same_num[1]] == b[same_num[0]]:
return True
if len(same_num) == 0:
return len(a) > len(set(a))
return False
```
提交结果: