力扣刷题记录#字符串#简单#859亲密字符串

题目描述

给定两个由小写字母构成的字符串 A 和 B ，只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果，就返回 true ；否则返回 false 。

示例

输入： A = “ab”, B = “ba”
输出： true

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

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

输入： A = “aaaaaaabc”, B = “aaaaaaacb”
输出： true

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

解答

根据示例总结的规律：

1. 两个字符串长度不等，返回false
2. 两个字符串完全相同：
   （1）A中无重复字符，返回false
   （2）A中有重复字符，可以交换重复字符，返回true
3. A和B等长，但部分字符不同，不同的字符对数小于等于2，交换一次即可变为相同，返回true

class Solution(object):
    def buddyStrings(self, A, B):
        """
        :type A: str
        :type B: str
        :rtype: bool
        """
        
        len_a = len(A)
        len_b = len(B)
        
        if len_a != len_b:
            return False
        
        diff_a = ""
        diff_b = ""
        for i in range(0,len_a):
            if A[i]!=B[i]:
                diff_a = diff_a + A[i]
                diff_b = diff_b + B[i]
        
        # A和B相同
        if len(diff_a)==0:
            # A中有重复元素，则可以交换重复元素
            if len(set(A))<len(A):
                return True
            else:
                return False
        
        # A和B中不同的对数超过2
        elif len(diff_a)>2:
            return False
        
        # A和B中不同的对数小于等于2
        else:
            # 交换A中元素与B相同
            if diff_a[::-1]==diff_b:
                return True
            else:
                return False