朋友圈 简单并查集

班上有 N 名学生。其中有些人是朋友,有些则不是。他们的友谊具备是传递性。若是已知 A 是 B 的朋友,B 是 C 的朋友,那么咱们能够认为 A 也是 C 的朋友。所谓的朋友圈,是指全部朋友的集合。bash

给定一个 N * N 的矩阵 M,表示班级中学生之间的朋友关系。若是Mi = 1,表示已知第 i 个和 j 个学生互为朋友关系,不然为不知道。你必须输出全部学生中的已知的朋友圈总数。ui

示例 1:spa

输入: 
[[1,1,0],
 [1,1,0],
 [0,0,1]]
输出: 2 
说明:已知学生0和学生1互为朋友,他们在一个朋友圈。
第2个学生本身在一个朋友圈。因此返回2。
复制代码

示例 2:code

输入: 
[[1,1,0],
 [1,1,1],
 [0,1,1]]
输出: 1
说明:已知学生0和学生1互为朋友,学生1和学生2互为朋友,因此学生0和学生2也是朋友,因此他们三个在一个朋友圈,返回1。
复制代码

注意:io

  1. N 在[1,200]的范围内。
  2. 对于全部学生,有Mi = 1。
  3. 若是有Mi = 1,则有Mj = 1。
class Solution {
    int[] f = new int[201];
    public int find( int x ){
        int p = x;
        while( p != f[ p ] ){
            p = f[ p ];
        }
        if( x != p ){
            while( x != p ){
                int t = f[ x ];
                f[ x ] = p ; 
                x = t;
            }
        }
        return p;
    }
    public void union(int  i , int j ){
        int p = find( i );
        int q = find( j );
        f[ q ] = f[ p ] ;
    }
    public int findCircleNum(int[][] M) {
       int n = M.length;
       for( int i = 0 ; i < n ; i++ ){
           f[ i ] = i ;
       } 
       for( int i = 0 ; i < n; i++ ){
           for( int j = 0; j < n ; j++ ){
               if( M[ i ][ j ] == 1 ){
                   union( i  , j );
               }
           }
       } 
       int res = 0;
       for( int i = 0 ; i < n ; i++ ){
           if( f[ i ] == i ){
               res++;
           }
       } 
        return res;
    }
}
复制代码
相关文章
相关标签/搜索