题目描述
给你长度相等的两个字符串 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
仅由小写英文字母组成来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/check-if-one-string-swap-can-make-strings-equal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目难度——简单
方法一:统计位置
根据题目来看,只有当两个字符串的不同位置的数量为2时才有可能满足要求,而且题目保证两个字符串长度相等,所以我们可以直接遍历。保存过程中不同位置的下标,最后对下标进行比较。
代码/Python
class Solution:
def areAlmostEqual(self, s1: str, s2: str) -> bool:
if s1 == s2:
return True
indx = []
for i in range(len(s1)):
if s1[i] != s2[i]:
indx.append(i)
if len(indx) != 2:
return False
i, j = indx[0], indx[1]
news = list(s1)
news[i], news[j] = news[j], news[i]
if ''.join(news) == s2:
return True
else:
return False
有点慢,能不能优化呢?上面用了一个列表来存储,有没有可能只用到两个常数变量来记录位置呢。以及最后,似乎没必要新建一个字符串然后交换字符位置然后再比较。直接比较两个下标位置上的字符是否相等就行了。
代码优化
class Solution:
def areAlmostEqual(self, s1: str, s2: str) -> bool:
if s1 == s2:
return True
x = y = -1
for i in range(len(s1)):
if s1[i] != s2[i]:
if x < 0:
x = i
elif y < 0:
y = i
else:
return False
if x != -1 and y != -1:
return s1[x] == s2[y] and s1[y] == s2[x]
else:
return False
也就快了4ms,可能是因为字符串最长也只有100吧。要是数量级更大一些,提速应该会更明显。
代码/C++
class Solution {
public:
bool areAlmostEqual(string s1, string s2) {
if(s1 == s2){
return true;
}
int x, y, i, length;
x = y = -1;
length = s1.size();
for(i = 0; i < length; i++){
if(s1[i] != s2[i]){
if(x < 0) x = i;
else if(y < 0) y = i;
else return false;
}
}
if(x != -1 && y != -1){
return s1[x] == s2[y] && s1[y] == s2[x];
}
else{
return false;
}
}
};
总结
一次遍历两个字符串,所以时间是O(N),只用到了常数变量,所以空间是O(1)。在优化前的空间是O(N)。