python 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

class Solution(object):
    def buddyStrings(self, A, B):
        """
        :type A: str
        :type B: str
        :rtype: bool
        """
        if A == B:
            if len(A) == len(set(A)):
                return False
            else:
                return True
        if len(A) != len(B):
            return False
        count = []
        for i in range(len(S)):
            if A[i] != B[i]:
                count.append(i)
        if len(count) != 2:
            return False
        if A[count[0]] == B[count[1]] and A[count[1]] == B[count[0]]:
            return True
        else:
            return False