解题思路:
(1)长度不等直接返回false
(2)如果重复字母大于等于3,返回false
(3)如果重复字母等于2,判断交换后,是否对应相等
(4)如果重复字母等于1,返回false
(5)如果字符串相等,判断是否A或B中是否存在相同的字母,有返回true,否则返回false
class Solution {
public:
bool buddyStrings(string A, string B) {
if (A.length()!=B.length()) { //长度不等直接返回false
return false;
}
vector<int > v; //存储不等下标
int count = 0; //存储不等个数
unordered_map<char,int> mymap;
for (int i=0;i<A.length();i++) {
mymap[A[i]]++; //判断重复字母个数
if (A[i]!=B[i]) {
v.push_back(i);
count++;
if (count==3) return false; //只能交换两个字母,超出则返回false
}
}
if (v.size()==1) return false; //只有一处不同也只能返回false
if (v.size()==0) {
for (auto w:mymap) {
if (w.second>=2) return true; //存在四个相同的字母
}
return false; //没有一个字母相同
}
if (A[v[0]]==B[v[1]] && A[v[1]]==B[v[0]]) return true;
else return false;
}
};