859.亲密字符串

给定两个由小写字母构成的字符串 A 和 B ，只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果，就返回 true ；否则返回 false 。
交换字母的定义是取两个下标 i 和 j （下标从 0 开始），只要 i!=j 就交换 A[i] 和 A[j] 处的字符。例如，在 “abcd” 中交换下标 0 和下标 2 的元素可以生成 “cbad” 。

示例 1：
输入： A = “ab”, B = “ba”
输出： true
解释： 你可以交换 A[0] = ‘a’ 和 A[1] = ‘b’ 生成 “ba”，此时 A 和 B 相等。

示例 2：
输入： A = “ab”, B = “ab”
输出： false
解释： 你只能交换 A[0] = ‘a’ 和 A[1] = ‘b’ 生成 “ba”，此时 A 和 B 不相等。

示例 3:
输入： A = “aa”, B = “aa”
输出： true
解释： 你可以交换 A[0] = ‘a’ 和 A[1] = ‘a’ 生成 “aa”，此时 A 和 B 相等。

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/buddy-strings
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

# 若a、b只有两个不同且则求出不同的元素进行交换并比较，若a、b不止两个不同则返回Flase，若a、b相同则判断其原长度和去重后的长度（若相等则表示无法交换）
class Solution:
    def buddyStrings(self, a: str, b: str) -> bool:
    	# 若a、b长度不相等则一定不符合题目要求
        if len(a) != len(b):
            return False

        dif = [i for i in range(len(a)) if a[i] != b[i]]  # 对比a、b，求出所有元素不同时的索引
        
        # 若只有两个元素不同则对a、b对应元素进行比较是否相等即可
        if len(dif) == 2 and a[dif[0]] == b[dif[1]] and a[dif[1]] == b[dif[0]]:
            return True

		# 若原长度和去重后长度相等则表示没有交换的余地
        if len(dif) == 0 and len(a) > len(set(a)):
            return True

        return False