leetcode 859. 亲密字符串(Buddy Strings)

Posted 2021-02-15 zhanzq1

目录

• 题目描述：
• 示例 1：
• 示例 2：
• 示例 3:
• 示例 4：
• 示例 5：
• 解法：

题目描述：

给定两个由小写字母构成的字符串 A 和 B ，只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果，就返回 true ；否则返回 false 。

示例 1：

输入： A = "ab", B = "ba"
输出： true

示例 2：

输入： A = "ab", B = "ab"
输出： false

示例 3:

输入： A = "aa", B = "aa"
输出： true

示例 4：

输入： A = "aaaaaaabc", B = "aaaaaaacb"
输出： true

示例 5：

输入： A = "", B = "aa"
输出： false

提示：

1. 0 <= A.length <= 20000
2. 0 <= B.length <= 20000
3. A 和 B 仅由小写字母构成。

---

解法：

class Solution {
public:
    bool buddyStrings(string A, string B) {
        if(A == B){
            vector<bool> flag(128, false);
            for(char ch : A){
                if(flag[ch]){
                    return true;
                }else{
                    flag[ch] = true;
                }
            }
            return false;
        }else if(A.size() != B.size()){
            return false;
        }else{
            int sz = A.size();
            int pre = -1;
            bool changed = false;   // when switch two charactors, set to true
            for(int i = 0; i < sz; i++){
                if(A[i] != B[i]){
                    if(pre == -1){
                        pre = i;
                    }else{
                        if(changed){
                            return false;
                        }
                        if(A[pre] != B[i] || A[i] != B[pre]){
                            return false;
                        }else{
                            changed = true;
                        }
                    }
                }
            }
            return changed;
        }
    }
};