No.104 - LeetCode839 - 联通图个数

ShellDawn

于 2019-09-12 12:54:28 发布

阅读量166

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/100768395

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种用于计算相似字符串组数量的算法。通过构建图结构，算法能有效地找出所有相似字符串组成的独立组，即使在字符串数量较大时也能保持良好的性能。相似性的判断基于两字符串之间的差异不超过两个字符。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    struct Edge{
        int from;
        int to;
        int pre;
        Edge(int a,int b,int c):from(a),to(b),pre(c){}
    };
    vector<Edge> E;
    int cnt;
    vector<int> pre;
    vector<int> vis;
    void addEdge(int from,int to){
        E.push_back(Edge(from,to,pre[from]));
        pre[from] = cnt++;
        E.push_back(Edge(to,from,pre[to]));
        pre[to] = cnt++;
    }
    void dfs(int S){
        vis[S] = 1;
        for(int i=pre[S];i!=0;i=E[i].pre){
            if(vis[E[i].to] == 0) dfs(E[i].to);
        }
    }
    bool judge(string& x,string& y){
        int cnt = 0;
        int N = x.length();
        for(int i=0;i<N;i++){
            if(x[i] != y[i]) cnt++;
            if(cnt > 2) return false;
        }
        return true;
    }
    int numSimilarGroups(vector<string>& A) {
        int N = A.size();
        int ans = 0;
        cnt = 1;
        E.push_back(Edge(0,0,0));
        pre.reserve(N);
        vis.reserve(N);
        for(int i=0;i<N;i++) pre[i] = vis[i] = 0;
        for(int i=0;i<N;i++){
            for(int j=i+1;j<N;j++){
                if(judge(A[i],A[j])) addEdge(i,j);
            }
        }
        for(int i=0;i<N;i++){
            if(vis[i] == 0){
                dfs(i);
                ans++;
            }
        }
        return ans;
    }
};