班上有 N 名學生。其中有些人是朋友,有些則不是。他們的友誼具有是傳遞性。如果已知 A 是 B 的朋友,B 是 C 的朋友,那么我們可以認為 A 也是 C 的朋友。所謂的朋友圈,是指所有朋友的集合。
給定一個 N * N 的矩陣 M,表示班級中學生之間的朋友關系。如果M[i][j] = 1,表示已知第 i 個和 j 個學生互為朋友關系,否則為不知道。你必須輸出所有學生中的已知的朋友圈總數。
示例 1:
輸入:
[[1,1,0],
[1,1,0],
[0,0,1]]
輸出:2
解釋:已知學生 0 和學生 1 互為朋友,他們在一個朋友圈。
第2個學生自己在一個朋友圈。所以返回 2 。
代碼
class Solution {int[] fa;public void init()//并查集的操作{for(int i=0;i<fa.length;i++)fa[i]=i;}public int find(int x){if(x!=fa[x])fa[x]=find(fa[x]);return fa[x];}public void union(int x,int y){x=find(x);y=find(y);if(x==y) return;fa[x]=y;}public int findCircleNum(int[][] M) {int n=M.length,res=0;fa=new int[n];init();for(int i=0;i<n;i++)//構建集合for(int j=0;j<n;j++){if(i==j) continue;if(M[i][j]==1)union(i,j);}Set<Integer> set=new HashSet<>();for(int i=0;i<n;i++)//查找不相交的集合{int fx=find(i);if(!set.contains(fx)){set.add(fx);res++;} }return res;}
}