哈希表题目：亲密字符串

题目

标题和出处

标题：亲密字符串

出处：859. 亲密字符串

难度

3 级

题目描述

要求

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

交换字母的定义是：取两个下标 $\text{i}$ 和 $\text{j}$（下标从 $\text{0}$ 开始）且满足 $\text{i}\ne \text{j}$，并交换 $\text{s[i]}$ 和 $\text{s[j]}$ 处的字符。

• 例如，在 $\text{"abcd"}$ 中交换下标 $\text{0}$ 和下标 $\text{2}$ 的元素可以生成 $\text{"cbad"}$。

示例

示例 1：

输入：$\text{s = "ab", goal = "ba"}$
输出：$\text{true}$
解释：你可以交换 $\text{s[0] = ‘a’}$ 和 $\text{s[1] = ‘b’}$ 生成 $\text{"ba"}$，此时 $\text{s}$ 和 $\text{goal}$ 相等。

示例 2：

输入：$\text{s = "ab", goal = "ab"}$
输出：$\text{false}$
解释：你只能交换 $\text{s[0] = ‘a’}$ 和 $\text{s[1] = ‘b’}$ 生成 $\text{"ba"}$，此时 $\text{s}$ 和 $\text{goal}$ 不相等。

示例 3：

输入：$\text{s = "aa", goal = "aa"}$
输出：$\text{true}$
解释：你可以交换 $\text{s[0] = ‘a’}$ 和 $\text{s[1] = ‘a’}$ 生成 $\text{"aa"}$，此时 $\text{s}$ 和 $\text{goal}$ 相等。

示例 4：

输入：$\text{s = "aaaaaaabc", goal = "aaaaaaacb"}$
输出：$\text{true}$

数据范围

• $\text{1}\le \text{s.length, goal.length}\le \text{2}\times {\text{10}}^{\text{4}}$
• $\text{s}$ 和 $\text{goal}$ 由小写英语字母组成

解法

思路和算法

由于交换字符串中的两个字母不会改变字符串的长度，因此如果可以通过交换 $s$ 中的两个字母得到 $\text{goal}$，则一定有 $s$ 和 $\text{goal}$ 的长度相同。如果 $s$ 和 $\text{goal}$ 的长度不同，则返回 $\text{false}$。

当 $s$ 和 $\text{goal}$ 的长度相同时，如果可以通过交换 $s$ 中的两个字母得到 $\text{goal}$，则交换的两个字母可能相同或不同。

• 如果交换的两个字母相同，则 $s$ 和 $\text{goal}$ 相等且至少有一个字母在同一个字符串中出现超过 $1$ 次。

• 如果交换的两个字母不同，则 $s$ 和 $\text{goal}$ 有两个下标处的字母不同，其余字母相同，假设字母不同的两个下标是 ${\text{index}}_1$ 和 ${\text{index}}_2$，${\text{index}}_1\ne {\text{index}}_2$，则有 $s[{\text{index}}_1]=\text{goal}[{\text{index}}_2]$ 且 $s[{\text{index}}_2]=\text{goal}[{\text{index}}_1]$。

根据上述分析，可以同时遍历 $s$ 和 $\text{goal}$，判断是否可以通过交换 $s$ 中的两个字母得到 $\text{goal}$。遍历过程中需要记录每个字母的出现次数、是否有重复字母、字母不同的下标个数和字母不同的下标。对于每个下标 $i$，进行如下操作：

1. 将 $s[i]$ 的出现次数加 $1$，如果该字母的出现次数大于 $1$，则有重复字母；

2. 如果 $s[i]\ne \text{goal}[i]$，则将字母不同的下标个数加 $1$，并更新字母不同的下标，如果字母不同的下标个数大于 $2$，则返回 $\text{false}$。

遍历结束之后，字母不同的下标个数一定不会大于 $2$（否则在遍历过程中就返回 $\text{false}$）。根据字母不同的下标个数，结果分别如下：

• 如果字母不同的下标个数是 $1$，则返回 $\text{false}$；

• 如果字母不同的下标个数是 $0$，则当有重复字母时返回 $\text{true}$，否则返回 $\text{false}$；

• 如果字母不同的下标个数是 $2$，则当 $s[{\text{index}}_1]=\text{goal}[{\text{index}}_2]$ 和 $s[{\text{index}}_2]=\text{goal}[{\text{index}}_1]$ 都成立时返回 $\text{true}$，否则返回 $\text{false}$。

代码

class Solution {
    public boolean buddyStrings(String s, String goal) {
        if (s.length() != goal.length()) {
            return false;
        }
        int[] counts = new int[26];
        boolean repeat = false;
        int length = s.length();
        int different = 0, index1 = -1, index2 = -1;
        for (int i = 0; i < length; i++) {
            char c1 = s.charAt(i), c2 = goal.charAt(i);
            counts[c1 - 'a']++;
            if (counts[c1 - 'a'] > 1) {
                repeat = true;
            }
            if (c1 != c2) {
                different++;
                if (different == 1) {
                    index1 = i;
                } else if (different == 2) {
                    index2 = i;
                } else {
                    return false;
                }
            }
        }
        if (different == 1) {
            return false;
        }
        if (different == 0) {
            return repeat;
        } else {
            return s.charAt(index1) == goal.charAt(index2) && s.charAt(index2) == goal.charAt(index1);
        }
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是字符串 $s$ 和 $\text{goal}$ 的长度。需要遍历两个字符串各一次，遍历过程中维护相关信息，遍历结束之后判断的时间是 $O(1)$。

• 空间复杂度：$O(|\Sigma |)$，其中 $\Sigma$ 是字符集，这道题中 $\Sigma$ 是全部小写英语字母，$|\Sigma |=26$。空间复杂度主要取决于哈希表，需要使用哈希表记录每个字母的出现次数。