给定两个由小写字母构成的字符串 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
提示:
0 <= A.length <= 20000
0 <= B.length <= 20000
A 和 B 仅由小写字母构成。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/buddy-strings
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
拿到这个题,我就想= =直接写,把所有条件A和B有一个不同,两个不同,三个不同都列出来就好啦。
class Solution {
public:
bool buddyStrings(string A, string B) {
vector<int> a;
if(A.size() != B.size()) return 0;
for(int i = 0;i<A.size();i++){
if(A[i] != B[i]) a.push_back(i);
}
//当有两个以上不同或一个不同
if(a.size() > 2) return 0;
//当有两个不同
else if(a.size() == 2 && (A[a[0]]==B[a[1]] && A[a[1]]==B[a[0]])) return 1;
//当没有不同
else if(a.empty()){
for(int i=0;i<A.size();i++)
if(count(A.begin(),A.end(),A[i])>1)
return 1;
}
return 0;
}
};