leetcode 亲密字符串

859. 亲密字符串

给你两个字符串 s 和 goal ，只要我们可以通过交换 s 中的两个字母得到与 goal 相等的结果，就返回 true ；否则返回 false 。

交换字母的定义是：取两个下标 i 和 j （下标从 0 开始）且满足 i != j ，接着交换 s[i] 和 s[j] 处的字符。

例如，在 “abcd” 中交换下标 0 和下标 2 的元素可以生成 “cbad” 。

示例 1：

1
2
3

输入：s = "ab", goal = "ba"
输出：true
解释：你可以交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba"，此时 s 和 goal 相等。

示例 2：

1
2
3

输入：s = "ab", goal = "ab"
输出：false
解释：你只能交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba"，此时 s 和 goal 不相等。

示例 3：

1
2
3

输入：s = "aa", goal = "aa"
输出：true
解释：你可以交换 s[0] = 'a' 和 s[1] = 'a' 生成 "aa"，此时 s 和 goal 相等。

示例 4：

1 2	输入：s = "aaaaaaabc", goal = "aaaaaaacb" 输出：true

提示：

1 <= s.length, goal.length <= 2 * 104
s 和 goal 由小写英文字母组成

思路

字符串长度不相等, 直接返回false
字符串相等的时候, 只要有重复的元素就返回true
A, B字符串有不相等的两个地方, 需要查看它们交换后是否相等即可.

参考答案

class Solution(object):
    def buddyStrings(self, s, goal):
        """
        :type s: str
        :type goal: str
        :rtype: bool
        """

        # 长度不同直接false
        if len(s) != len(goal): return False
        
        # 由于必须交换一次，在相同的情况下，交换相同的字符
        if s == goal and len(set(goal)) < len(goal): return True
        
        # 使用 zip 进行匹配对比，挑出不同的字符对
        dif = [(a, b) for a, b in zip(s, goal) if a != b]
        
        # 对数只能为2，并且对称，如 (a,b)与(b,a)
        return len(dif) == 2 and dif[0] == dif[1][::-1]

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