【LeetCode】1790. 仅执行一次字符串交换能否使两个字符串相等

题目

1790. 仅执行一次字符串交换能否使两个字符串相等

给你长度相等的两个字符串 s1 和 s2 。一次 字符串交换 操作的步骤如下：选出某个字符串中的两个下标（不必不同），并交换这两个下标所对应的字符。

如果对 其中一个字符串 执行 最多一次字符串交换 就可以使两个字符串相等，返回 true ；否则，返回 false 。

示例 1：

输入：s1 = "bank", s2 = "kanb"
输出：true
解释：例如，交换 s2 中的第一个和最后一个字符可以得到 "bank"

示例 2：

输入：s1 = "attack", s2 = "defend"
输出：false
解释：一次字符串交换无法使两个字符串相等

示例 3：

输入：s1 = "kelb", s2 = "kelb"
输出：true
解释：两个字符串已经相等，所以不需要进行字符串交换

示例 4：

输入：s1 = "abcd", s2 = "dcba"
输出：false

提示：

• 1 <= s1.length, s2.length <= 100
• s1.length == s2.length
• s1 和 s2 仅由小写英文字母组成

思路

• 若两个字符串长度不相等，则一定不能通过交换一次字符使其相等
• 遍历字符串，若第一次出现不相等，保存两边对应的字符
• 若仅仅出现一次不相等，返回false
• 若第二次出现不相等，判断是否与第一次出现的相同，是则标记交换，否则返回true
• 若出现多次不相等，则直接返回false

代码

class Solution:
    def areAlmostEqual(self, s1: str, s2: str) -> bool:
        a,b = "",""
        exchange = False
        if len(s1) != len(s2): return False
        for i in range(len(s1)):
            if s1[i] == s2[i]: continue
            if s1[i] != s2[i] and not exchange and a=="":
                a,b = s1[i],s2[i]
            elif s1[i] != s2[i] and not exchange and s1[i]==b and s2[i] == a:
                exchange = True
            else: return False
        return False if a!="" and not exchange else True

复杂度

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$