859. 亲密字符串

208 阅读2分钟

这是我参与11月更文挑战的第21天,活动详情查看:2021最后一次更文挑战

859. 亲密字符串

给你两个字符串 s 和 goal ,只要我们可以通过交换 s 中的两个字母得到与 goal 相等的结果,就返回 true ;否则返回 false 。

交换字母的定义是:取两个下标 i 和 j (下标从 0 开始)且满足 i != j ,接着交换 s[i] 和 s[j] 处的字符。

例如,在 "abcd" 中交换下标 0 和下标 2 的元素可以生成 "cbad" 。  

示例 1:

输入:s = "ab", goal = "ba"
输出:true
解释:你可以交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba",此时 s 和 goal 相等。

示例 2:

输入:s = "ab", goal = "ab"
输出:false
解释:你只能交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba",此时 s 和 goal 不相等。

示例 3:

输入:s = "aa", goal = "aa"
输出:true
解释:你可以交换 s[0] = 'a' 和 s[1] = 'a' 生成 "aa",此时 s 和 goal 相等。

示例 4:

输入:s = "aaaaaaabc", goal = "aaaaaaacb"
输出:true

提示:

  • 1 <= s.length, goal.length <= 2 * 10410^4
  • s 和 goal 由小写英文字母组成

解题思路

分为三种情况进行讨论

  1. 第一种就是满足亲密字符串定义的,字符串s和字符串goal之间恰好有两个位置的字符是不相同的,而且我们恰好可以通过交换字符串s这两个下标的元素使得二者相等
  2. 第二种情况就是字符串s和字符串goal只有一个位置上的字符是不相同的,因此我们不存在与其交换的下标,因此这种情况应该返回false
  3. 第三种情况就是两个字符串是完全相等的,但是他们存在两个不同下标的字符是相同的,因此我们可以通过交换这两个下标,达到不改变原字符串,使两字符串相等的目的。

代码

class Solution {
public:
    bool buddyStrings(string s, string goal) {

        int n = s.size();
        int dif1 = -1;

        for (int i = 0; i < n; ++i) {
            if (s[i] != goal[i]) {
                if (dif1 == -1)
                    dif1 = i;
                else {
                    swap(goal[i], goal[dif1]);
                    break;
                }
            }
        }

        if (dif1 == -1) {
            sort(s.begin(), s.end());
            for (int j = 1; j < n; ++j) {
                if (s[j] == s[j - 1])
                    return true;
            }
            return false;

        } else
            return s==goal;
    }
};