Question

1657. 确定两个字符串是否接近

English Version

题目描述

如果可以使用以下操作从一个字符串得到另一个字符串，则认为两个字符串 接近 ：

• 操作 1：交换任意两个 现有 字符。
  • 例如，abcde -> aecdb
• 操作 2：将一个 现有 字符的每次出现转换为另一个 现有 字符，并对另一个字符执行相同的操作。
  • 例如，aacabb -> bbcbaa（所有 a 转化为 b ，而所有的 b 转换为 a ）

你可以根据需要对任意一个字符串多次使用这两种操作。

给你两个字符串，word1 和 word2 。如果_ _word1_ _和_ _word2_ _接近 ，就返回 true ；否则，返回_ _false_ _。

 

示例 1：

输入：word1 = "abc", word2 = "bca"
输出：true
解释：2 次操作从 word1 获得 word2 。
执行操作 1："abc" -> "acb"
执行操作 1："acb" -> "bca"

示例 2：

输入：word1 = "a", word2 = "aa"
输出：false
解释：不管执行多少次操作，都无法从 word1 得到 word2 ，反之亦然。

示例 3：

输入：word1 = "cabbba", word2 = "abbccc"
输出：true
解释：3 次操作从 word1 获得 word2 。
执行操作 1："cabbba" -> "caabbb"
执行操作 2："caabbb" -> "baaccc"
执行操作 2："baaccc" -> "abbccc"

提示：

• 1 <= word1.length, word2.length <= 105
• word1 和 word2 仅包含小写英文字母

解法

方法一：计数 + 排序

根据题目描述，两个字符串接近，需要同时满足以下两个条件：

1. 字符串 word1 和 word2 包含的字母种类必须相同；
2. 将字符串 word1 和 word2 的所有字符出现次数排序，得到的两个数组必须相同。

因此，我们可以先用数组或哈希表分别统计 word1 和 word2 中每种字母出现的次数，然后比较两者是否相同，不相同则提前返回 false。

否则，我们将对应的次数排序，然后依次比较对应位置的两个次数是否相同，不同则返回 false。

遍历结束，返回 true。

时间复杂度 $O(m + n + C \times \log C)$，空间复杂度 $O(C)$。其中 $m$ 和 $n$ 分别为字符串 word1 和 word2 的长度，而 $C$ 是字母种类。本题中 $C=26$。

class Solution:
  def closeStrings(self, word1: str, word2: str) -> bool:
    cnt1, cnt2 = Counter(word1), Counter(word2)
    return sorted(cnt1.values()) == sorted(cnt2.values()) and set(
      cnt1.keys()
    ) == set(cnt2.keys())

class Solution {
  public boolean closeStrings(String word1, String word2) {
    int[] cnt1 = new int[26];
    int[] cnt2 = new int[26];
    for (int i = 0; i < word1.length(); ++i) {
      ++cnt1[word1.charAt(i) - 'a'];
    }
    for (int i = 0; i < word2.length(); ++i) {
      ++cnt2[word2.charAt(i) - 'a'];
    }
    for (int i = 0; i < 26; ++i) {
      if ((cnt1[i] == 0) != (cnt2[i] == 0)) {
        return false;
      }
    }
    Arrays.sort(cnt1);
    Arrays.sort(cnt2);
    return Arrays.equals(cnt1, cnt2);
  }
}

class Solution {
public:
  bool closeStrings(string word1, string word2) {
    int cnt1[26]{};
    int cnt2[26]{};
    for (char& c : word1) {
      ++cnt1[c - 'a'];
    }
    for (char& c : word2) {
      ++cnt2[c - 'a'];
    }
    for (int i = 0; i < 26; ++i) {
      if ((cnt1[i] == 0) != (cnt2[i] == 0)) {
        return false;
      }
    }
    sort(cnt1, cnt1 + 26);
    sort(cnt2, cnt2 + 26);
    return equal(cnt1, cnt1 + 26, cnt2);
  }
};

func closeStrings(word1 string, word2 string) bool {
  cnt1 := make([]int, 26)
  cnt2 := make([]int, 26)
  for _, c := range word1 {
    cnt1[c-'a']++
  }
  for _, c := range word2 {
    cnt2[c-'a']++
  }
  if !slices.EqualFunc(cnt1, cnt2, func(v1, v2 int) bool { return (v1 == 0) == (v2 == 0) }) {
    return false
  }
  sort.Ints(cnt1)
  sort.Ints(cnt2)
  return slices.Equal(cnt1, cnt2)
}

function closeStrings(word1: string, word2: string): boolean {
  const cnt1 = Array(26).fill(0);
  const cnt2 = Array(26).fill(0);
  for (const c of word1) {
    ++cnt1[c.charCodeAt(0) - 'a'.charCodeAt(0)];
  }
  for (const c of word2) {
    ++cnt2[c.charCodeAt(0) - 'a'.charCodeAt(0)];
  }
  for (let i = 0; i < 26; ++i) {
    if ((cnt1[i] === 0) !== (cnt2[i] === 0)) {
      return false;
    }
  }
  cnt1.sort((a, b) => a - b);
  cnt2.sort((a, b) => a - b);
  return cnt1.join('.') === cnt2.join('.');
}

impl Solution {
  pub fn close_strings(word1: String, word2: String) -> bool {
    let mut cnt1 = vec![0; 26];
    let mut cnt2 = vec![0; 26];
    for c in word1.chars() {
      cnt1[((c as u8) - b'a') as usize] += 1;
    }
    for c in word2.chars() {
      cnt2[((c as u8) - b'a') as usize] += 1;
    }
    for i in 0..26 {
      if (cnt1[i] == 0) != (cnt2[i] == 0) {
        return false;
      }
    }
    cnt1.sort();
    cnt2.sort();
    cnt1 == cnt2
  }
}