文章目录
- 题目描述
- 题目难度——简单
- 方法一:统计位置
- 代码/Python
- 代码优化
- 代码/C++
- 总结
题目描述
给你长度相等的两个字符串 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)。
