leetcode721. 合并账户【两种方法;并查集;dfs】

文章目录


并查集(方法一)

cpp 复制代码
class Solution {
    unordered_map<string, int> index; // 每个邮箱都有一个唯一编号
    int root[10010];                  // 并查集

    void unionset(int a, int b) {
        int rt1 = findrt(a), rt2 = findrt(b);
        if (rt1 == rt2) return; // 很重要
        root[rt1] += root[rt2];
        root[rt2] = rt1;
    }

    int findrt(int a) {
        if (root[a] < 0) return a;
        return root[a] = findrt(root[a]);
    }

public:
    vector<vector<string>> accountsMerge(vector<vector<string>>& accounts) {
        memset(root, -1, sizeof(root));
        // 给出index
        for (int i = 0; i < accounts.size(); i++) {
            for (int j = 1; j < accounts[i].size(); j++) {
                index[accounts[i][j]]++;
            }
        }
        int email_cnt = 0;
        for (auto it = index.begin(); it != index.end(); it++) {
            it->second = email_cnt++;
        }

        // 并
        for (int i = 0; i < accounts.size(); i++) {
            for (int j = 2; j < accounts[i].size(); j++) {
                unionset(index[accounts[i][j - 1]], index[accounts[i][j]]);
            }
        }
        // 查
        unordered_map<int, int> rt_and_ansIndex;
        vector<vector<string>> ans;

        for (int i = 0; i < accounts.size(); i++) {
            int rt_index = findrt(index[accounts[i][1]]);
            if (rt_and_ansIndex.count(rt_index) == 0) {
                rt_and_ansIndex[rt_index] = ans.size();
                ans.push_back(vector<string>());
                ans.back().push_back(accounts[i][0]); // 账户名
            }

            for (int j = 1; j < accounts[i].size(); j++) {
                ans[rt_and_ansIndex[rt_index]].push_back(accounts[i][j]); // 邮箱
            }
        }

        for (int i = 0; i < ans.size(); i++) {
            sort(ans[i].begin() + 1, ans[i].end()); // 排序
            ans[i].erase(unique(ans[i].begin() + 1, ans[i].end()), ans[i].end()); // 去重
        }
        return ans;
    }
};

dfs(方法二)

cpp 复制代码
class Solution {
    unordered_map<string, int> index; // 每个邮箱都有一个唯一编号
    string emails[10010];
    vector<vector<int>> g; // 图,邻接表存法
    bool visited[10010];
    vector<vector<string>> ans;
    
    void dfs(int n) {
        visited[n] = true;
        ans.back().push_back(emails[n]);
        for (int i = 0; i < g[n].size(); i++) {
            if (!visited[g[n][i]]) {
                dfs(g[n][i]);
            }
        }
        return;
    }

public:
    vector<vector<string>> accountsMerge(vector<vector<string>>& accounts) {
        memset(visited, false, sizeof(visited));
        // 给出index
        for (int i = 0; i < accounts.size(); i++) {
            for (int j = 1; j < accounts[i].size(); j++) {
                index[accounts[i][j]]++;
            }
        }
        int email_cnt = 0;
        for (auto it = index.begin(); it != index.end(); it++) {
            it->second = email_cnt++;
            emails[it->second] = it->first;
        }
        g.resize(email_cnt);
        // 建图
        for (int i = 0; i < accounts.size(); i++) {
            for (int j = 2; j < accounts[i].size(); j++) {
                g[index[accounts[i][1]]].push_back(index[accounts[i][j]]);
                g[index[accounts[i][j]]].push_back(index[accounts[i][1]]); //双向边
            }
        }

        // dfs
        for (int i = 0; i < accounts.size(); i++) {
            if (!visited[index[accounts[i][1]]]) {
                ans.push_back(vector<string>());
                ans.back().push_back(accounts[i][0]);
                dfs(index[accounts[i][1]]);
            }
        }

        for (int i = 0; i < ans.size(); i++) {
            sort(ans[i].begin() + 1, ans[i].end()); // 排序
            ans[i].erase(unique(ans[i].begin() + 1, ans[i].end()),
                         ans[i].end()); // 去重
        }
        return ans;
    }
};

dfs换一种写法

cpp 复制代码
class Solution {
    unordered_map<string, vector<string>> g; // 图,邻接表存法,存邮箱,而不是
    set<string> visited;
    vector<vector<string>> ans;
    void dfs(string str) {
        visited.insert(str);
        ans.back().push_back(str);
        for (int i = 0; i < g[str].size(); i++) {
            if (visited.count(g[str][i]) == 0) {
                dfs(g[str][i]);
            }
        }
        return;
    }

public:
    vector<vector<string>> accountsMerge(vector<vector<string>>& accounts) {
        // 建图
        for (int i = 0; i < accounts.size(); i++) {
            for (int j = 2; j < accounts[i].size(); j++) {
                g[accounts[i][1]].push_back(accounts[i][j]);
                g[accounts[i][j]].push_back(accounts[i][1]);
            }
        }

        // dfs
        for (int i = 0; i < accounts.size(); i++) {
            if (visited.count(accounts[i][1]) == 0) {
                ans.push_back(vector<string>());
                ans.back().push_back(accounts[i][0]);
                dfs(accounts[i][1]);
            }
        }

        for (int i = 0; i < ans.size(); i++) {
            sort(ans[i].begin() + 1, ans[i].end()); // 排序
            ans[i].erase(unique(ans[i].begin() + 1, ans[i].end()), ans[i].end()); // 去重
        }
        return ans;
    }
};

如果unordered_map改为map,更慢。

相关推荐
算AI15 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
懒羊羊大王&16 小时前
模版进阶(沉淀中)
c++
owde16 小时前
顺序容器 -list双向链表
数据结构·c++·链表·list
GalaxyPokemon16 小时前
Muduo网络库实现 [九] - EventLoopThread模块
linux·服务器·c++
W_chuanqi16 小时前
安装 Microsoft Visual C++ Build Tools
开发语言·c++·microsoft
hyshhhh16 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
tadus_zeng17 小时前
Windows C++ 排查死锁
c++·windows
EverestVIP17 小时前
VS中动态库(外部库)导出与使用
开发语言·c++·windows
杉之17 小时前
选择排序笔记
java·算法·排序算法
烂蜻蜓17 小时前
C 语言中的递归:概念、应用与实例解析
c语言·数据结构·算法