1. 程式人生 > 實用技巧 >547. 朋友圈-並查集-中等

547. 朋友圈-並查集-中等

問題描述

班上有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 。
示例 2:

輸入:
[[1,1,0],
[1,1,1],
[0,1,1]]
輸出:1
解釋:已知學生 0 和學生 1 互為朋友,學生 1 和學生 2 互為朋友,所以學生 0 和學生 2 也是朋友,所以他們三個在一個朋友圈,返回 1 。

提示:

1 <= N <= 200
M[i][i] == 1
M[i][j] == M[j][i]

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/friend-circles

解答

//並查集
class Solution {
    int[] parent;
    public int find(int x){
        if(parent[x] == -1)return x;
        return find(parent[x]);
    }
    public void union(int x, int y){//結為盟友
int xParent = find(x); int yParent = find(y); if(xParent != yParent) parent[xParent] = yParent; } public int findCircleNum(int[][] M) { int n=M.length; if(n==0)return 0; parent = new int[n]; Arrays.fill(parent, -1); for(int i=0;i<n;i++){
for(int j=0;j<n;j++){ if(M[i][j] == 1)union(i,j); } } int circle = 0; for(int i:parent) if(i==-1)circle++; return circle; } }