力扣721.账户合并
-
并查集 + dfs
- 哈希表以地址为下标 存账户下标数组
- 遍历每个账户下标
- 其中的每个email如果还对应其他下标(说明两下标对应的账户名称一样 可以合并)
- 那么继续dfs
-
class Solution { public: vector<vector<string>> accountsMerge(vector<vector<string>>& accounts) { unordered_map<string,vector<int>> email_to_idx; for(int i=0;i<accounts.size();i++) for(int k=1;k<accounts[i].size();k++) email_to_idx[accounts[i][k]].push_back(i); unordered_set<string> email_set; //记录每个下标是否遍历过 vector<int> vis(accounts.size()); auto dfs = [&](auto &&dfs,int i) ->void { vis[i] = true; //遍历其中每个地址 for(int k=1;k<accounts[i].size();k++) { string &email = accounts[i][k]; if(email_set.contains(email)) continue; //没有放进来过 就插入 并看是否有其他下标也包含这个地址 email_set.insert(email); for(int j:email_to_idx[email]) if(!vis[j]) dfs(dfs,j); } }; vector<vector<string>> res; for(int i=0;i<vis.size();i++) { //dfs过就continue if(vis[i]) continue; email_set.clear(); dfs(dfs,i); //账户名称 + 账户地址的集合(字典序排序) 放入答案 vector<string> cur = {accounts[i][0]}; cur.insert(cur.end(),email_set.begin(),email_set.end()); sort(cur.begin()+1,cur.end()); res.push_back(cur); } return res; } };