Leetcode刷题--冗余连接

LeetCode-684 冗余连接

题目

输入一个图，该图由一个有着N个节点 (节点值不重复1, 2, …, N) 的树及一条附加的边构成。附加的边的两个顶点包含在1到N中间，这条附加的边不属于树中已存在的边。

结果图是一个以边组成的二维数组。每一个边的元素是一对[u, v] ，满足 u < v，表示连接顶点u 和v的无向图的边。

返回一条可以删去的边，使得结果图是一个有着N个节点的树。如果有多个答案，则返回二维数组中最后出现的边。答案边 [u, v] 应满足相同的格式 u < v。

	输入: [[1,2], [1,3], [2,3]]
输出: [2,3]
解释: 给定的无向图为:
  1
 / \
2 - 3

思路

本题是对查并集的考察，先来学习一下查并集的知识。
CSDN上有一篇对查并集详细解释的博客，通俗易懂，这里贴以下链接
查并集
其实查并集简单来讲，就好比一个摸个地区有一个规模庞大的家族。结果导致大家找对象的时候犯难了，生怕跟对方有亲戚关系。
有一天，隔壁的二狗看上了邻村的阿美，想去跟他谈对象，但是要先查一下家谱

int find(int x){ //二狗开始查家谱
        //二狗不识字，找对门的秀才帮他查
        int r = x;
        //pre[x]表示x的上一级，也就是二狗他爹
        //秀才就一直找，知道找到二狗的老祖宗为止
        while (r != pre[r])
            r = pre[r];
          
        //秀才看二狗脑子不灵光，记不住这么多辈分，干脆直接把二狗祖宗名字
        //直接告诉二狗，叫他记好(也就是路径压缩)
        int tmp1 = x;
        int tmp2 = 0;
        if (tmp1 != r){
            //暂存当前状态的上级
            tmp2 = pre.get(r);
            pre.set(tmp1, tmp2);
            tmp1 = tmp2;
        }
        return r;
    }

经过一趟find，二狗就能知道自己的祖宗是谁了，同样的阿美也要find一下自己的祖宗。

 boolean join(int x, int y){
 		//两人各找各的祖宗
        int x_root = find(x);
        int y_root = find(y);
        //如果祖宗不相同，就可以开心地过没羞没臊的生活了
        if (x_root != y_root){
        	//心机的阿美说：二狗你要娶我可以，但你需要把你的祖宗的名字
        	//归在我的祖宗下面。
        	//二狗觉得以后都是一家人了，就傻呵呵的同意了
            pre[x_root] = y_root;
            return true;
        }
        //有情人终成兄妹啊
        return false;
    }

经过join之后，两家原本没关系，现在就联系起来了。

Java语言实现

讲了这么多终于开始做题了。

1.遍历所有的边edges，将连通的结点放入同一个集合，形成一个联通分量GG。
2.在遍历的过程中，如果边(a, b)(a,b)的两个结点aa, bb已经属于同一联通分量，
则(a, b)(a,b)就是该联通分量的冗余边。

class Solution {
    ArrayList<Integer> pre = new ArrayList<>();
    public int[] findRedundantConnection(int[][] edges) {
        int row = edges.length;
        for (int i = 0; i < 1001; i++){
            pre.add(i);
        }
        for (int i = 0; i < row; i++){
            boolean flag = join(edges[i][0], edges[i][1]);
            if (!flag)
                return edges[i];
        }
        return null;
    }
    int find(int x){
        //工具人r
        int r = x;
        //一直找到根节点为止
        while (r != pre.get(r))
            r = pre.get(r);
        //路径压缩
        int tmp1 = x;
        int tmp2 = 0;
        if (tmp1 != r){
            //暂存当前状态的上级
            tmp2 = pre.get(r);
            pre.set(tmp1, tmp2);
            tmp1 = tmp2;
        }
        return r;
    }

    boolean join(int x, int y){
        int x_root = find(x);
        int y_root = find(y);
        if (x_root != y_root){
            pre.set(x_root, y_root);
            return true;
        }
        return false;
    }
}

时间复杂度约为：O(n)
空间复杂度：O(n)