【LeetCode】C++ ：简单题 - 字符串 859. 亲密字符串

859. 亲密字符串

难度简单126

给定两个由小写字母构成的字符串 

A

 和 

B

 ，只要我们可以通过交换 

A

 中的两个字母得到与 

B

 相等的结果，就返回 

true

 ；否则返回 

false

 。

交换字母的定义是取两个下标 

i

 和 

j

 （下标从   开始），只要 

i!=j

 就交换 

A[i]

 和 

A[j]

 处的字符。例如，在 

"abcd"

 中交换下标   和下标 

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
      

提示：

1. 0 <= A.length <= 20000

2. 0 <= B.length <= 20000

3. 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];
        }


    }
};