LeetCode 859. 亲密字符串

示例 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 <= 20000
0 <= B.length <= 20000
A 和 B 仅由小写字母构成。

class Solution {
public:
    bool buddyStrings(string A, string B) {
        if(A.size() < 2 || B.size() < 2 || A.size() != B.size())
        	return false;
        int countA[26] = {0}, countB[26] = {0}, diff = 0;
        for(int i = 0; i < A.size(); ++i)
        {
        	if(A[i] != B[i])
        		diff++;
            countA[A[i]-'a']++;
            countB[B[i]-'a']++;
        }
        for(int i = 0; i < 26; ++i)
            if(countA[i] != countB[i])
                return false;//个数不一样。不行
        bool hasSameChar = false;
        for(int i = 0; i < 26; ++i)
        	if(countA[i] >= 2)
        	{
        		hasSameChar = true;//有相同的字符
        		break;
        	}
        if(diff == 2 || (diff == 0 && hasSameChar))
        	return true;//不同的位置为2，或者全部一样，有相同字符可以互换
        return false;
    }
};