LeetCode 684. Redundant Connection [Medium]

最新推荐文章于 2023-07-10 21:11:19 发布

YuMingJing_

最新推荐文章于 2023-07-10 21:11:19 发布

阅读量639

点赞数

分类专栏：算法分析与设计文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/YuMingJing_/article/details/78514234

版权

算法分析与设计专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种使用并查集（UnionFind）解决无向图中寻找冗余连接问题的方法。通过构建并查集数据结构，可以高效地检测并删除构成环的最后一条边。

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

原题地址

题目内容

这里写图片描述

题目分析

题目的意思大致为，给了一个无向图，让我们删掉组成环的最后一条边。一开始想的做法是想采用拓扑排序来判断是否由环，如果有再删去多余的边。后面看了leetcode的discuss，发现他用了一个较为简单的方法，union find。建立一个长数组，来存储每个节点，初始化为-1，每加入一条新的边（u，v)，都通过find()函数来查找u之前是否有一条到V的路，如果有，那么这条边就是多余的，如果没有就添加到root数组里面,root[u] = v,就代表有一条u->v的路径。

代码实现

class Solution {
public:
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        vector<int> root(20001,-1);
        for(auto i:edges){
            int a = find(root,i[0]);
            int b = find(root,i[1]);
            if(a == b){
                return i;
            }
            else{
                root[a] = b;
            }
        }

    }
    int find(vector<int>&root,int x){
        while(root[x] != -1){
            x = root[x];
        }
        return x;
    }
};