LeetCode-力扣-1790-仅执行一次字符串交换能否使两个字符串相等(带我的代码与官方代码对比)-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.使用count计数的时候就应该想到vector有size()这个函数，还不用自己维护。
2.当新引入两个下标进行计数的时候就应该想到直接引入一个新的容器使用0和1两个地址来存储。
综上应该使用Vector
but我这样比官方的空间要好一点嘿嘿

class Solution {
public:
    bool areAlmostEqual(string s1, string s2) {
        if(s1.length() != s2.length()){//不等长直接false
            return false;
        }
        if(s1 == s2){//相等直接true
            return true;
        }
        int count = 0;//不匹配的度数
        int n = s1.length();
        int pos1 = -1, pos2 = -1;//两个不匹配的位置
        for(int i = 0; i < n; i++){
            if(s1[i] != s2[i]){
                if(pos1 == -1){
                    pos1 = i;
                }
                else{
                    pos2 = i;
                }
                count++;
            }
        }
        if(pos1 == -1 || pos2 == -1){
            return false;
        }
        if(s1[pos1] == s2[pos2] && s1[pos2] == s2[pos1] && count == 2){
            return true;
        }
        else{
            return false;
        }
        return false;
    }
};

官方题解

class Solution {
public:
    bool areAlmostEqual(string s1, string s2) {
        int n = s1.size();
        vector<int> diff;
        for (int i = 0; i < n; ++i) {
            if (s1[i] != s2[i]) {
                if (diff.size() >= 2) {//这里的>=让我想了几分钟，可以看上一行代码，又发现了一对新的不匹配
                                       //如果等于2说明之前已经有两对了，那么现在有三对，是不可以的。
                    return false;
                }
                diff.emplace_back(i);
            }
        }
        if (diff.size() == 0) {
            return true;
        }
        if (diff.size() != 2) {
            return false;
        }
        return s1[diff[0]] == s2[diff[1]] && s1[diff[1]] == s2[diff[0]];
    }
};