给定两个由小写字母构成的字符串 A 和
B ,只要我们可以通过交换 A
中的两个字母得到与 B 相等的结果,就返回
true ;否则返回 false 。
示例 1:
输入: A = "ab", B = "ba"
输出: true
示例 2:
输入: A = "ab", B = "ab"
输出: false
示例 3:
输入: A = "aa", B = "aa"
输出: true
示例 4:
输入: A = "aaaaaaabc", B = "aaaaaaacb"
输出: true
示例 5:
输入: A = "", B = "aa"
输出: false
提示:
0 <= A.length <= 200000 <= B.length <= 20000A 和 B 仅由小写字母构成。 bool buddyStrings(string A, string B) {
if (A.size() != B.size()) return false;
int len = A.size(), index = 0;
unordered_set<char> res;
string strA, strB;
for (int i = 0; i < len; ++i) {
if (A[i] != B[i]) {
strA += A[i];
strB += B[i];
}
else
res.insert(A[i]);
}
if (res.size() == 1) return true;
if (strA.size() != 2) return false;
return strA[0] == strB[1] && strA[1] == strB[0];
}执行用时 :8 ms, 在所有 cpp 提交中击败了68.31%的用户
内存消耗 :9.2 MB, 在所有 cpp 提交中击败了19.33%的用户
就三种情况
1、长度不一样或者长度小于2,直接false
2、不匹配的个数超过2个,false
3、如果全部一样,则看A中是否有重复的字符,有就是true了,
否则就看两个不匹配的位序上的字符交换后是否一样
bool buddyStrings(string A, string B) {
if (A.size() != B.size() || A.size()<2) return false;
int len = A.size();
vector<int> res;
res.reserve(len);
for (int i = 0; i < len; ++i) {
if (A[i] != B[i]) {
res.push_back(i);
if (res.size() > 2) return false;
}
}
if (res.size() == 0) {
unordered_set<char> misMatch(A.begin(), A.end());
return misMatch.size() < len;
}
return A[res[0]] == B[res[1]] && A[res[1]] == B[res[0]];
}