class Solution {
public:
int numSimilarGroups(vector<string>& strs) {
}
};
839. 相似字符串组
如果交换字符串 X 中的两个不同位置的字母，使得它和字符串 Y 相等，那么称 X 和 Y 两个字符串相似。如果这两个字符串本身是相等的，那它们也是相似的。
例如，"tars" 和 "rats" 是相似的 (交换 0 与 2 的位置)； "rats" 和 "arts" 也是相似的，但是 "star" 不与 "tars"，"rats"，或 "arts" 相似。
总之，它们通过相似性形成了两个关联组：{"tars", "rats", "arts"} 和 {"star"}。注意，"tars" 和 "arts" 是在同一组中，即使它们并不相似。形式上，对每个组而言，要确定一个单词在组中，只需要这个词和该组中至少一个单词相似。
给你一个字符串列表 strs。列表中的每个字符串都是 strs 中其它所有字符串的一个字母异位词。请问 strs 中有多少个相似字符串组？
示例 1：
输入：strs = ["tars","rats","arts","star"] 输出：2
示例 2：
输入：strs = ["omv","ovm"] 输出：1
提示：
1 <= strs.length <= 3001 <= strs[i].length <= 300strs[i] 只包含小写字母。strs 中的所有单词都具有相同的长度，且是彼此的字母异位词。原站题解
java 解法, 执行用时: 10 ms, 内存消耗: 40.7 MB, 提交时间: 2023-01-12 10:08:38
class Solution {
int[] f;
public int numSimilarGroups(String[] strs) {
int n = strs.length;
int m = strs[0].length();
f = new int[n];
for (int i = 0; i < n; i++) {
f[i] = i;
}
for (int i = 0; i < n; i++) {
for (int j = i + 1; j < n; j++) {
int fi = find(i), fj = find(j);
if (fi == fj) {
continue;
}
if (check(strs[i], strs[j], m)) {
f[fi] = fj;
}
}
}
int ret = 0;
for (int i = 0; i < n; i++) {
if (f[i] == i) {
ret++;
}
}
return ret;
}
public int find(int x) {
return f[x] == x ? x : (f[x] = find(f[x]));
}
public boolean check(String a, String b, int len) {
int num = 0;
for (int i = 0; i < len; i++) {
if (a.charAt(i) != b.charAt(i)) {
num++;
if (num > 2) {
return false;
}
}
}
return true;
}
}
javascript 解法, 执行用时: 100 ms, 内存消耗: 43.4 MB, 提交时间: 2023-01-12 10:08:10
/**
* @param {string[]} strs
* @return {number}
*/
var numSimilarGroups = function(strs) {
const n = strs.length;
const m = strs[0].length;
const f = new Array(n).fill(0).map((element, index) => index);
for (let i = 0; i < n; i++) {
for (let j = i + 1; j < n; j++) {
const fi = find(i), fj = find(j);
if (fi === fj) {
continue;
}
if (check(strs[i], strs[j], m)) {
f[fi] = fj;
}
}
}
let ret = 0;
for (let i = 0; i < n; i++) {
if (f[i] === i) {
ret++;
}
}
return ret;
function find(x) {
return f[x] === x ? x : (f[x] = find(f[x]));
}
function check(a, b, len) {
let num = 0;
for (let i = 0; i < len; i++) {
if (a[i] !== b[i]) {
num++;
if (num > 2) {
return false;
}
}
}
return true;
}
};
golang 解法, 执行用时: 8 ms, 内存消耗: 4 MB, 提交时间: 2023-01-12 10:05:15
type unionFind struct {
parent, size []int
setCount int // 当前连通分量数目
}
func newUnionFind(n int) *unionFind {
parent := make([]int, n)
size := make([]int, n)
for i := range parent {
parent[i] = i
size[i] = 1
}
return &unionFind{parent, size, n}
}
func (uf *unionFind) find(x int) int {
if uf.parent[x] != x {
uf.parent[x] = uf.find(uf.parent[x])
}
return uf.parent[x]
}
func (uf *unionFind) union(x, y int) {
fx, fy := uf.find(x), uf.find(y)
if fx == fy {
return
}
if uf.size[fx] < uf.size[fy] {
fx, fy = fy, fx
}
uf.size[fx] += uf.size[fy]
uf.parent[fy] = fx
uf.setCount--
}
func (uf *unionFind) inSameSet(x, y int) bool {
return uf.find(x) == uf.find(y)
}
func isSimilar(s, t string) bool {
diff := 0
for i := range s {
if s[i] != t[i] {
diff++
if diff > 2 {
return false
}
}
}
return true
}
func numSimilarGroups(strs []string) int {
n := len(strs)
uf := newUnionFind(n)
for i, s := range strs {
for j := i + 1; j < n; j++ {
if !uf.inSameSet(i, j) && isSimilar(s, strs[j]) {
uf.union(i, j)
}
}
}
return uf.setCount
}
python3 解法, 执行用时: 244 ms, 内存消耗: 15.3 MB, 提交时间: 2023-01-12 10:04:52
'''
我们把每一个字符串看作点，字符串之间是否相似看作边，那么可以发现本题询问的是给定的图中有多少连通分量。
于是可以想到使用并查集维护节点间的连通性。
我们枚举给定序列中的任意一对字符串，检查其是否具有相似性，如果相似，那么我们就将这对字符串相连。
在实际代码中，我们可以首先判断当前这对字符串是否已经连通，如果没有连通，
我们再检查它们是否具有相似性，可以优化一定的时间复杂度的常数。
'''
class Solution:
def numSimilarGroups(self, strs: List[str]) -> int:
n = len(strs)
f = list(range(n))
def find(x: int) -> int:
if f[x] == x:
return x
f[x] = find(f[x])
return f[x]
def check(a: str, b: str) -> bool:
num = 0
for ac, bc in zip(a, b):
if ac != bc:
num += 1
if num > 2:
return False
return True
for i in range(n):
for j in range(i + 1, n):
fi, fj = find(i), find(j)
if fi == fj:
continue
if check(strs[i], strs[j]):
f[fi] = fj
ret = sum(1 for i in range(n) if f[i] == i)
return ret