LeetCode859. 亲密字符串

给定两个由小写字母构成的字符串 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(object):
    def buddyStrings(self, A, B):
        """
        :type A: str
        :type B: str
        :rtype: bool
        """
        i = 0
        if len(A) == 0 or len(B) == 0 or len(A) != len(B): 
            return False
        n = len(A)
        A,B = list(A),list(B)
        # different记录AB中不同的字符的索引
        different = []
        while i<n:
            if A[i] != B[i]:
                different.append(i)
            i+=1
        # 如果不同的字符大于两个，直接返回False，因为只能交换一次
        if len(different) > 2:
            return False
        # 如果没有不同字符，只要有一个字符数量大于1，则是亲密字符串
        # (因为这个时候可以交换任意字符数量大于等于2的字符)
        elif len(different) == 0:
            for a in set(A):
                if A.count(a)>1:
                    return True
            return False
        else:
            A[different[0]],A[different[1]] = A[different[1]],A[different[0]]
            return A==B