Python Leetcode(859.亲密字符串)

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

示例 5：

输入： A = “”, B = “aa”
输出： false

提示：

0 <= A.length <= 20000
0 <= B.length <= 20000
A 和 B 仅由小写字母构成。

Solution:(把字符串转换成列表进行操作，先进行排序，如果两个列表不相等则返回False；相等则需要判断两个字符串对应位置不相等的个数，如果为2则返回True；如果为0，即两个字符串完全相等，需要判断是否有重复的元素，没有则返回False；个数如果为其它值则返回False)

class Solution(object):
    def buddyStrings(self, A, B):
        """
        :type A: str
        :type B: str
        :rtype: bool
        """
        a = list(A)
        b = list(B)
        a.sort()
        b.sort()
        
        if a != b:
            return False
        
        num = 0
        for i in range(len(A)):
            if A[i] != B[i]:
                num += 1
        
        if num == 2:
            return True
        elif num == 0:
            if len(set(A)) == len(A):
                return False
            else:
                return True
        else:
            return False

solution = Solution()
print(solution.buddyStrings("ab", "ba"))

True