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


class Solution:
    def buddyStrings(self, A: str, B: str) -> bool:
        if len(A) != len(B):
            return 0
        
        index = []
        
        for i in range(len(A)):
            if A[i] != B[i]:
                index.append(i)
                
        if len(index) == 0:
            dicts = {}
            for j in A:
                if j not in dicts.keys():
                    dicts[j] = 1
                else:
                    dicts[j] += 1
            for item in dicts.values():
                if item>1:
                    return 1
            return 0
        elif len(index) == 2:
            a = A[:index[0]]+A[index[1]]+A[index[0]+1:index[1]]+A[index[0]]+A[index[1]+1:]
            if a == B:
                return 1
            else:
                return 0
        else:
            return 0