题目描述
给你两个字符串 s 和 goal ，只要我们可以通过交换 s 中的两个字母得到与 goal 相等的结果，就返回 true ；否则返回 false 。
交换字母的定义是：取两个下标 i 和 j （下标从 0 开始）且满足 i != j ，接着交换 s[i] 和 s[j] 处的字符。
例如，在 “abcd” 中交换下标 0 和下标 2 的元素可以生成 “cbad” 。
Example1:
输入：s = “ab”, goal = “ba” 输出：true 解释：你可以交换 s[0] = ‘a’ 和 s[1] = ‘b’ 生成 “ba”，此时 s 和 goal 相等。
Example2:
输入：s = “ab”, goal = “ab” 输出：false 解释：你只能交换 s[0] = ‘a’ 和 s[1] = ‘b’ 生成 “ba”，此时 s 和 goal 不相等。
Example3:
输入：s = “aa”, goal = “aa” 输出：true 解释：你可以交换 s[0] = ‘a’ 和 s[1] = ‘a’ 生成 “aa”，此时 s 和 goal 相等。
Example4:
输入：s = “aaaaaaabc”, goal = “aaaaaaacb” 输出：true
Note:
- 每个字符串仅由字符 ‘0’ 或 ‘1’ 组成。
- 1 <= a.length, b.length <= 10^4
- 字符串如果不是 “0” ，就都不含前导零。
解题思路
class Solution {
public:
bool buddyStrings(string s, string goal) {
if(s.size() != goal.size()){
return false;
}
int first = -1, second = -1;
if(s == goal){
vector<int> count(26);
for(auto& c : s){
count[c - 'a']++;
if(count[c - 'a'] > 1){
return true;
}
}
return false;
}else{
for(int i = 0; i < s.size(); i++){
if(s[i] != goal[i]){
if(first == -1){
first = i;
}else if(second == -1){
second = i;
}else{
return false;
}
}
}
}
return second != -1 && s[first] == goal[second] && s[second] == goal[first];
}
};
