题目描述
给你两个字符串 s 和 goal ,只要我们可以通过交换 s 中的两个字母得到与 goal 相等的结果,就返回 true ;否则返回 false 。
交换字母的定义是:取两个下标 i 和 j (下标从 0 开始)且满足 i != j ,接着交换 s[i] 和 s[j] 处的字符。
例如,在 “abcd” 中交换下标 0 和下标 2 的元素可以生成 “cbad” 。
代码
class Solution {
public:
bool buddyStrings(string a, string b) {
if(a.size() != b.size()) return false;
if(a==b)
{
unordered_map<char, int> cnt;
for(auto c:a)
if(++cnt[c]>1)
return true;
return false;
}
vector<int> q;
for(int i=0;i<a.size();i++)
if(a[i]!=b[i])
q.push_back(i);
if(q.size()!=2) return false;
int x=q[0], y=q[1];
if(a[x]==b[y] && a[y]==b[x]) return true;
return false;
}
};