亲密字符串

发表于 2018-08-27 更新于 2021-09-11 分类于 LeetCode 阅读次数：本文字数： 886 阅读时长 ≈ 1 分钟

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

示例 1：

1 2	输入： A = "ab", B = "ba" 输出： true

示例 2：

1 2	输入： A = "ab", B = "ab" 输出： false

示例 3:

1 2	输入： A = "aa", B = "aa" 输出： true

示例 4：

1 2	输入： A = "aaaaaaabc", B = "aaaaaaacb" 输出： true

示例 5：

1 2	输入： A = "", B = "aa" 输出： false

class Solution {
public:
    bool buddyStrings(string A, string B) {
        if (A.size() != B.size()) return false;
        if (A == B) {
            vector<int> count(26, 0);
            for (int i = 0; i < A.size(); ++i)
                count[A[i] - 'a']++;

            for (int c: count)
                if (c > 1) return true;
            return false;
        } else {
            int first = -1, second = -1;
            for (int i = 0; i < A.size(); ++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]);
        }
    }
};