leetcode721——Accounts Merge

最新推荐文章于 2023-05-16 17:25:14 发布

tzyshiwolaogongya

最新推荐文章于 2023-05-16 17:25:14 发布

阅读量852

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构-并查集

本文链接：https://blog.youkuaiyun.com/tzyshiwolaogongya/article/details/79586900

数据结构-并查集专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用并查集算法解决邮箱账户合并的问题。通过初始化每个邮箱为独立节点，并进行查找与合并操作，最终实现同一账户下所有邮箱的统一管理。

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

题目大意：给出一些账户和账户下的邮箱，将同一账户的邮箱合并

分析：并查集应用。将邮箱作为并查集中的点，初始化每个邮箱是自己的父节点，然后查找合并，最后按格式输出。注意find函数做了路径压缩，可以提高查找效率。

代码：

class Solution {  
public:  
    vector<vector<string>> accountsMerge(vector<vector<string>>& accounts) {  
        map<string, string> owner;          // map from the 邮箱 to 名字  
        map<string, string> parents;        // map from an 邮箱 to its 邮箱的父亲 
        map<string, set<string>> unions;    // the unions of accounts  并查集string下的结点们
        for (int i = 0; i < accounts.size(); ++i) {  //初始化
            for (int j = 1; j < accounts[i].size(); ++j) {  
                parents[accounts[i][j]] = accounts[i][j];   // initialize as itself  
                owner[accounts[i][j]] = accounts[i][0];  
            }  
        }  
        for (int i = 0; i < accounts.size(); ++i) {         // find and union 查找与合并 
            string ancestor = find(accounts[i][1], parents);  
            for (int j = 2; j < accounts[i].size(); ++j) {  
                parents[find(accounts[i][j], parents)] = ancestor;  
            }  
        }  
        for (int i = 0; i < accounts.size(); ++i) {  //将每个邮箱结点放入它的根节点带领的并查集中
            for (int j = 1; j < accounts[i].size(); ++j) {  
                unions[find(accounts[i][j], parents)].insert(accounts[i][j]); 
            }  
        }  
        vector<vector<string>> res;  
        for (auto p : unions) {  //取出每一个map对
            vector<string> emails(p.second.begin(), p.second.end());  
            emails.insert(emails.begin(), owner[p.first]);  //在emails.begin()前面插入owner[p.first]元素
            res.push_back(emails);  
        }  
        return res;  
    }  
private:  
    string find(string s, map<string, string> &p) {  
        if(p[s] != s) p[s] = find(p[s],p);
        return p[s];  
    }  
};