难度简单126
给定两个由小写字母构成的字符串
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
仅由小写字母构成。
这题有待下次自己写一遍思路和代码,觉得官方题解这个很厉害
class Solution {
public:
bool buddyStrings(string A, string B) {
if(A.length() != B.length()){
return false;
}
if(A == B){
vector<int> count(26);
for(auto ch: A){
count[ch-'a']++;
}
for(auto i: count){
if(i > 1){
return true;
}
}
return false;
}else{
int first = -1, second = -1;
for(int i = 0; i < A.length(); i++){
if(A[i] != B[i]){
if(first == -1){
first = i;
}else if(second == -1){
second = i;
}else{
return false;
}
}
}
return second != -1 && A[first] == B[second] && A[second] == B[first];
}
}
};