【力扣刷题】仅执行一次字符串交换能否使两个字符串相等

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

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

一开始以为要用分治法，来递归一下，差点要大改代码。其实只要避免重复交换就可以了，需要一个列表来标记交换过的位置。其实就是像看到的那样，用双指针一次循环找过去就行了。

我的代码：

class Solution(object):
    def areAlmostEqual(self, s1, s2):
        """
        :type s1: str
        :type s2: str
        :rtype: bool
        """
        if len(s1) != len(s2):
            return False
        if s1 == s2:
            return True
        
        i = j =0
        count = 0
        change_index = []
        while i < len(s1) and j < len(s2):
            if s1[i] != s2[j] and i not in change_index:
                change = False
                for k in range(j+1, len(s2)):
                    if s2[k] == s1[i] and s2[i] == s1[k]:
                        count += 1
                        change_index.append(k)
                        change = True
                if not change:
                    return False
            i += 1
            j += 1
        if count == 1:
            return True
        else:
            return False