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