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 仅由小写字母构成。

执行结果

代码描述

思路：分几种情况考虑：

1. 长度不相等，肯定不满足。
2. 两个空字符串，肯定不满足。
3. 其中有两处不一样，进行转移，看转移后是否一样。
4. 其中两个字符串相等，则看字符串中，是否有1个字符出现次数大于等于2次，这样能满足2次的交换。
5. 其他情况，count > 2, 均为false。

class Solution {
public:
    bool buddyStrings(string A, string B) {
        if(A.size() != B.size())    return false;        // 长度不等，肯定不等
        if(A.size() == 0 && B.size() == 0)  return false;    // 都为空，没法转移2次，故false
        if(A == B && isTwo(A))    return true;  // 当两个字符串本身就相等时，需要判断是否有一个字符出现过两次。
        int count = 0;
        vector<int> vec;
        for(int i = 0; i < A.size(); ++i)
        {
            if(A[i] != B[i])
            {
                ++count;
                vec.push_back(i);
            }
        }
        if(count == 2)        // 当有两次不相等时，交换之后，判断是否相等。
        {
            swap(A[vec[0]], A[vec[1]]);
            if(A == B)
                return true;
            else
                return false;
        }
        return false;
    }
    
    // 判断一个字符是否出现过大于等于2次。
    bool isTwo(string str)
    {
        for(int i = 0; i < str.size()-1; ++i)
        {
            for(int j = i+1; j < str.size(); ++j)
            {
                if(str[i] == str[j])
                    return true;
            }
        }
        return false;
    }
};