给你两个字符串 s 和 goal ,只要我们可以通过交换 s 中的两个字母得到与 goal 相等的结果,就返回 true ;否则返回 false 。
交换字母的定义是:取两个下标 i 和 j (下标从 0 开始)且满足 i != j ,接着交换 s[i] 和 s[j] 处的字符。
例如,在 "abcd" 中交换下标 0 和下标 2 的元素可以生成 "cbad" 。
示例 1:
输入:s = "ab", goal = "ba"
输出:true
解释:你可以交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba",此时 s 和 goal 相等。
示例 2:
输入:s = "ab", goal = "ab"
输出:false
解释:你只能交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba",此时 s 和 goal 不相等。
示例 3:
输入:s = "aa", goal = "aa"
输出:true
解释:你可以交换 s[0] = 'a' 和 s[1] = 'a' 生成 "aa",此时 s 和 goal 相等。
示例 4:
输入:s = "aaaaaaabc", goal = "aaaaaaacb"
输出:true
提示:
1 <= s.length, goal.length <= 2 * 104
s 和 goal 由小写英文字母组成
#include <iostream>
#include <string>
#include <unordered_map>
using namespace std;
class Solution {
public:
bool buddyStrings(string s, string goal) {
int count = 0;
unordered_map<char, int> hashMp1;
unordered_map<char, int> hashMp2;
bool isBigThanTwo = false;
if (s.length() != goal.length()) {
return false;
}
for (int i = 0; i < s.size(); i++) {
if (hashMp1.count(s[i]) == 0) {
hashMp1[s[i]] = 1;
}
else {
hashMp1[s[i]]++;
}
if (hashMp2.count(goal[i]) == 0) {
hashMp2[goal[i]] = 1;
}
else {
hashMp2[goal[i]]++;
}
if (s[i] != goal[i]) {
count++;
}
}
unordered_map<char, int>::iterator it = hashMp1.begin();
while (it != hashMp1.end()) {
if (hashMp1[it->first] > 1) {
isBigThanTwo = true;
}
if (hashMp1[it->first] != hashMp2[it->first]) {
return false;
}
it++;
}
if (isBigThanTwo && s == goal) {
return true;
}
if (count == 2) {
return true;
}
return false;
}
};
int main() {
Solution* ps = new Solution();
cout << ps->buddyStrings("ab", "ba");
return 0;
}