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
思路
并查集: 贴出并查集的代码:
int par[max_n]; //父亲数组
int rank[max_n]; //树的高度
void init(int n) {
//初始化
for (int i = 0; i < n; i++) {
par[i] = i;
rank[i] = 0;
}
}
int find(int x) {
//查询树的根
if (par[x] == x) return x;
else {
return par[x] = find(par[x]);
}
}
void unite(int x, int y) {
//合并x,y所属的集合
x = find(x);
y = find(y);
if (x == y) return;
if (rank[x] < rank[y]) par[x] = y;
else {
par[y] = x;
if (rank[x] == rank[y]) rank[x]++;
}
}
bool same(int x, int y) {
return find(x) == find(y);
}
**
代码:答案
**
class Solution {
vector<int> ans; //辅助数组
bool check(string& s1,string& s2,int n){
//检查函数
int sum=0;
for(int i=0;i<n;i++){
if(s1[i]!=s2[i]) sum++;
if(sum>2) return false;
}
return true;
}
int find(int n){
//查找函数
return ans[n]==n?n:ans[n]=find(ans[n]);
}
public:
int numSimilarGroups(vector<string>& a) {
int m=a.size();
int n=a[0].size();
ans.resize(m);
for(int i=0;i<m;i++){
ans[i]=i;
}
for(int i=0;i<m;i++){
for(int j=i+1;j<m;j++){
int f1=find(i);
int f2=find(j);
if(f1==f2) continue;
if(check(a[i],a[j],n)) {
ans[f1]=f2; //合并,
}
}
}
int count=0;
for(int i=0;i<m;i++){
if(ans[i]==i) count++;
}
return count;
}
};