Graph Valid Tree
题目地址:
https://leetcode.com/problems/graph-valid-tree/#/description
题目:
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.
For 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.
解题思路:
这道题主要就是用union find的思想来解。需要注意的是n个vertex只能有n-1条边,不然多余n-1的话一定有环。
代码:
public boolean validTree(int n, int[][] edges) { 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 edges.length == n - 1; } private int find(int[] nums, int i){ if(nums[i] == -1){ return i; } return find(nums, nums[i]); }

Comments
Post a Comment