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

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

力扣题目链接：https://leetcode.cn/problems/check-if-one-string-swap-can-make-strings-equal/

给你长度相等的两个字符串 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

遍历一遍字符串，用一个数组diff把两个字符串对应位置字母不同处的下标记录下来。

如果diff为空，那就说明两个字符串原本就相同，直接返回true

否则就得进行一次交换

一次交换能交换好的前提是diff中存放的“不同元素”个数为2，且其中一个字符串这两个位置交换后和另一个字符串相等。

如果满足上述前提，就返回true，否则返回false

• 时间复杂度$O(n)$，其中$n$是字符串长度
• 空间复杂度$O(Diff)$，其中$Diff$是两个字符串中，对应位置元素不同的个数。这里经过优化可以将复杂度降为$O(1)$，因为最多有两个不同元素的位置，因此如果diff中已经存在了两个元素，并且又遇到了不对应的元素，就直接返回false

AC代码

C++

class Solution {
public:
    bool areAlmostEqual(string s1, string s2) {
        if (s1.size() != s2.size())
            return false;
        vector<int> diff;
        for (int i = 0; i < s1.size(); i++) {
            if (s1[i] != s2[i]) {
                diff.push_back(i);
            }
        }
        if (diff.empty())
            return true;
        if (diff.size() == 2 && (s1[diff[0]] == s2[diff[1]] && s1[diff[1]] == s2[diff[0]]))
            return true;
        return false;
    }
};

方法二：费力不讨好的优化

与方法一类似，方法一中，我们知道最多有两个位置元素不同。因此我们可以不使用动态数组，而是直接使用两个变量来解决。这样，理论上开销应该会小一丢丢。

用两个变量firstDiffLoc和secondDiffLoc分别记录第一个和第二个元素不同的位置。二者初始值都是-1

因此，要判断已经有几个不同元素的位置，只需要判断这两个变量是否为-1即可。

• 时间复杂度$O(n)$，其中$n$是字符串长度
• 空间复杂度$O(1)$

因本题数据量只有100，因此上述方法是“费力不讨好系列”，效果并不明显，甚至时间占用增加了。

AC代码

C++

class Solution {
public:
    bool areAlmostEqual(string& s1, string& s2) {
        if (s1.size() != s2.size())
            return false;
        int firstDiffLoc = -1, secondDiffLoc = -1;
        for (int i = s1.size() - 1; i >= 0; i--) {
            if (s1[i] != s2[i]) {
                if (firstDiffLoc == -1) {
                    firstDiffLoc = i;
                }
                else if (secondDiffLoc == -1) {
                    secondDiffLoc = i;
                }
                else {
                    return false;
                }
            }
        }
        if (firstDiffLoc == -1)  // 全相同
            return true;
        if (secondDiffLoc != -1 && (s1[firstDiffLoc] == s2[secondDiffLoc] && s1[secondDiffLoc] == s2[firstDiffLoc]))  // 正好两个不同，且不同的那两个互换后相同了
            return true;
        return false;
    }
};

同步发文于CSDN，原创不易，转载请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/127258599