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

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

给你长度相等的两个字符串 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 仅由小写英文字母组成

解

方法：一次遍历
同时遍历两个字符串同一位置，设置一个变量cnt表示出现不同的字符的次数，有如下情况

• 两个字符完全相同，此时cnt = 0
• 有一次字符不同，此时cnt = 1，返回false
• 有两处字符不相同且交换后符合条件，此时cnt = 2
• 有两处字符不相同且交换后不符合条件，返回false
• 有大于两处的字符不同，返回false

class Solution {
public:
    bool areAlmostEqual(string s1, string s2) {
        int len = s1.size();
        int cnt = 0;
        char c1, c2;
        for (int i = 0; i < len; ++i) {
            if (s1[i] == s2[i]) {
                continue;
            }
            if (0 == cnt) {
                c1 = s1[i];
                c2 = s2[i];
            } else if (1 == cnt) {	
                if (c1 != s2[i] || c2 != s1[i]) {	// 是否能交换
                    return false;
                }
            } else {
                return false;
            }
            ++cnt;
        }
        
        return cnt == 0 || cnt == 2;            
    }
};