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

源码：

class Solution {
    public boolean buddyStrings(String A, String B) {
        // 第一种情况：两个字符串长度不等
        if (A.length() != B.length()) {
            return false;
        }
        // 第二种情况：两个字符串相同
        // 接着判断 A 中是否存在两个以上相同的字符
        // 如果存在直接返回 true 否则 返回 false
        if (A.equals(B)) {
            Map<Character, Integer> map = new HashMap<>();
            for (char x : A.toCharArray()) {
                map.put(x, map.getOrDefault(x, 0) + 1);
            }
            for (int j = 0; j < A.length(); j++) {
                if (map.get(A.charAt(j)) > 1) {
                    return true;
                }
            }
            return false;
        }
        // 第三种情况：判断两个字符串不相同的部分有几处
        // 如果有两处再判断将两个字符反过来，结果会不会和 B 相同
        // 如果存在1处或者2处以上的不相同部分，那么直接返回 false
        int count = 0;
        char A1 = 'a';
        char B1 = 'a';
        char A2 = 'a';
        char B2 = 'a';
        for (int i = 0; i < A.length(); i++) {
            if (A.charAt(i) != B.charAt(i)) {
                count++;
                if (count == 1) {
                    A1 = A.charAt(i);
                    B1 = B.charAt(i);
                }
                if (count == 2) {
                    A2 = A.charAt(i);
                    B2 = B.charAt(i);
                }
            }
            if (count > 2) {
                return false;
            }
        }
        return count == 2 && A1 == B2 && A2 == B1;
    }
}