Leetcode 859 亲密字符串

题目描述：

给定两个由小写字母构成的字符串 A 和 B ，只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果，就返回 true ；否则返回 false 。

 

示例 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

 

提示：

1. 0 <= A.length <= 20000
2. 0 <= B.length <= 20000
3. A 和 B 仅由小写字母构成。

思路：这题其实挺恶心的，不能漏情况...，长度不相等或者长度小于2，则直接返回false，如果两个字符串相等，就判断字符串中有没有相等的字符。有就true，没有就false。 如果两个字符串不相等，就看不相等的字符串是否为2个，如果为2个且排序完成后两个字符串相等，就返回true,其余情况返回false。

代码：

class Solution {
public:
    bool buddyStrings(string A, string B) {
        
        if(A.length() != B.length() || A.length() < 2)
            return false;
        
        int flag = 0;
        for(int i = 0; i < A.length(); i++)
        {
            for(int j = i + 1; j < A.length(); j++)
            {
                if(A[i] == A[j])
                {
                    flag = 1;
                    break;
                }

            }
        }
                
        if(flag == 1 && A == B)
            return true;
        
        
        int sum = 0;

        for(int i = 0; i < A.length(); i++)
        {
            if(A[i] != B[i])
            {
                sum++;
                if(sum > 2)
                    return false;
            }

        }
        
        if(sum == 2)
        {
            sort(A.begin(), A.end());
            sort(B.begin(), B.end());
            
            return A == B;
        }

        
        return false;
        
    }
};