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

提示：

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

二、代码和思路

1.首先遍历查看A,B对应位置不相等的字符，如果超过二个则return False

2.若对应位置不相等的字母数为0，则查看字符串中是否有相等字母；

3.若数量为2，则需查看对应位置交换后是否相等

class Solution(object):
    def buddyStrings(self, A, B):
        """
        :type A: str
        :type B: str
        :rtype: bool
        """
        m,n=len(A),len(B)
        if m != n:return False
        diff_word=[];diff_num=0
        for i in range(m):
            if A[i] != B[i]:
                diff_num += 1
                diff_word.append(A[i])
                diff_word.append(B[i])
                if diff_num>2:
                    return False          
        if diff_num==0:
            for i in range(m):
                if A[i] in A[i+1:]:
                    return True
        if diff_num==2 and (diff_word[0] == diff_word[3]) and (diff_word[1]==diff_word[2]):
            return True
        return False
            

三、运行结果