給定從 0 到 n-1?標號的?n 個結點,和一個無向邊列表(每條邊以結點對來表示),請編寫一個函數用來判斷這些邊是否能夠形成一個合法有效的樹結構。
示例 1:
輸入: n = 5, 邊列表 edges = [[0,1], [0,2], [0,3], [1,4]]
輸出: true
示例 2:
輸入: n = 5, 邊列表 edges = [[0,1], [1,2], [2,3], [1,3], [1,4]]
輸出: false
注意:你可以假定邊列表 edges 中不會出現重復的邊。由于所有的邊是無向邊,邊?[0,1]?和邊 [1,0]?是相同的,因此不會同時出現在邊列表 edges 中。
思路:并查集,沒有環(find(x)!=find(y)),并且根=1(根多了就是森林了),我們認為是一顆樹。
class Solution {int[] parent;//這是記錄關系的數組//查找int find(int parent[], int i) {if (parent[i] == -1)return i;return find(parent, parent[i]);}//合并void union(int parent[], int x, int y) {int xset = find(parent, x);int yset = find(parent, y);if (xset != yset)parent[xset] = yset;}public boolean validTree(int n, int[][] edges) {int len=edges.length;parent = new int[n];Arrays.fill(parent, -1);for (int i = 0; i < len; i++) {if(find(parent,edges[i][0])==find(parent,edges[i][1]))return false;union(parent, edges[i][0], edges[i][1]);}int count = 0;//查根的數量for (int i = 0; i < n; i++)if (parent[i] == -1)count++;return count==1;}
}
?