[LeetCode] Graph Valid Tree [Union Find]

104828720 发布于2019-08-14 15:10 / 2654人阅读

Problem

Given n nodes labeled from 0 to n - 1 and a list of undirected edges (each edge is a pair of nodes), write a function to check whether these edges make up a valid tree.

Example

Given n = 5 and edges = [[0, 1], [0, 2], [0, 3], [1, 4]], return true.
Given n = 5 and edges = [[0, 1], [1, 2], [2, 3], [1, 3], [1, 4]], return false.

Note

You can assume that no duplicate edges will appear in edges. Since all edges are undirected, [0, 1] is the same as [1, 0] and thus will not appear together in edges.

Solution

public class Solution {
    
    int[] parents;
    public boolean validTree(int n, int[][] edges) {
        if (n-1 != edges.length) {
            return false;
        }
        parents = new int[n];
        //initialize n nodes as each own parent
        for (int i = 0; i < n; i++) {
            parents[i] = i;
        }
        for (int i = 0; i < n-1; i++) {
            if (find(edges[i][0]) == find(edges[i][1])) {
                return false; //if two nodes on edge[i] have the same parent, this edge makes a circle causing it invalid
            }
            else union(edges[i][0], edges[i][1]); //else union the two nodes on edge[i]
        }
        return true; 
    }
    public int find(int node) {
        //find the very first parent node, which is when the parent is the node itself
        if (parents[node] == node) {
            return node;
        }
        //find parent recursively
        return find(parents[node]);
    }
    public void union(int a, int b) {
        int finda = parents[a], findb = parent[b];
        //when node a and node b have different ancient parents, union them with the same parent
        if (finda != findb) {
            parents[finda] = findb;
        }
    }
}

Update 2018-10

class Solution {
    public boolean validTree(int n, int[][] edges) {
        if (edges.length != n-1) return false;
        int[] nums = new int[n];
        Arrays.fill(nums, -1);
        for (int i = 0; i < edges.length; i++) {
            int x = find(nums, edges[i][0]);
            int y = find(nums, edges[i][1]);
            if (x == y) return false;
            nums[x] = y;
        }
        return true;
    }
    private int find(int[] nums, int k) {
        if (nums[k] == -1) return k;
        else return find(nums, nums[k]);
    }
}

GPU云服务器云服务器 union-find graph LeetCode graph编程语言

文章版权归作者所有，未经允许请勿转载,若此文章存在违规行为，您可以联系管理员删除。

转载请注明本文地址：https://www.ucloud.cn/yun/65399.html

[Leetcode] Graph Valid Tree 图与树

摘要：这样就将一个集合的节点归属到同一个集合号下了。最后如果并查集中只有一个集合，则说明可以构建树。 Graph Valid Tree Given n nodes labeled from 0 to n - 1 and a list of undirected edges (each edge is a pair of nodes), write a function to check w...

luqiuwen 2019-08-14 12:36 评论0 收藏0
[Leetcode] Graph Valid Tree 判断一个图是否为树

摘要：只有一个连通分量还没有环，就是树，无根树。无向图边的两端是对称的，无向图讲究连通这个概念，没有方向，没有拓扑，很像集合，所以非常适合用并查集来解决。并查集写法参考注意方法用找的老大哥，合并的是的老大哥。 Graph Valid Tree Given n nodes labeled from 0 to n - 1 and a list of undirected edges (each...

xbynet 2019-08-14 14:04 评论0 收藏0
261. Graph Valid Tree

摘要：题目链接检查图的连通性及是否有环，可以，，从一个点出发看能不能遍历所有的点，同时来检查是否有环。还可以用检查是否有环，最后看的数量是否等于来判断是不是。 261. Graph Valid Tree 题目链接：https://leetcode.com/problems... 检查图的连通性及是否有环，可以dfs，bfs，从一个点出发看能不能遍历所有的点，同时visited来检查是否有环。...

Jinkey 2019-08-16 10:29 评论0 收藏0
[LeetCode] 684. Redundant Connection

Problem In this problem, a tree is an undirected graph that is connected and has no cycles. The given input is a graph that started as a tree with N nodes (with distinct values 1, 2, ..., N), with one...

lncwwn 2019-08-16 13:55 评论0 收藏0
[LeetCode] 399. Evaluate Division

Problem Equations are given in the format A / B = k, where A and B are variables represented as strings, and k is a real number (floating point number). Given some queries, return the answers. If the ...

BlackMass 2019-08-16 13:40 评论0 收藏0