并查集

题目描述

某社交APP有很多用户，其中有些人会互相关注。互相关注的用户我们认为是朋友，朋友具有传递性。如果已知A是B的朋友，B是C的朋友，那么A也是C的朋友。朋友之间组成朋友圈。

给定一个N*N的矩阵M，表示小红书的互相关注关系。M[i][j]=1表示第i个用户和第j个用户相互关注，M[i][j]=0表示单向关注或无关注。请输出所有用户中朋友圈的数量。

输入：

第一行为N，表示用户总数，接下来N行是每个用户的相互关注信息，每行N个数字，由空格分隔的0或1组成，表示用户跟其他用户的关系。

输出：

整数K，表示符合条件的小红书的数量。

样例输入：

3

1 1 0

1 1 0

0 0 1 

样例输出：

2

 

 基本思路

• 这是一道典型的并查集题目，关于并查集的解法网上有很多。本文尝试使用树的广度遍历来解决此类问题
• 将N个用户及其之间的关系看成图，本质是有多少个独立的图
• 从每个用户出发，进行树的广度优先遍历。如果能通过该用户访问到其他用户，且其他用户未被访问过，说明该用户与其他用户组成了一个朋友圈

代码上实现 

package bishi.xiaohongshu;

import java.util.LinkedList;
import java.util.Scanner;

/**
 * @基本功能:并查集
 * @program:summary
 * @author:peicc
 * @create:2019-09-14 14:55:31
 **/
public class Main1_1 {
    static int N;
    static int[][] e;
    static int[] book;
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        N=sc.nextInt();//用户数量
        e=new int[N][N];
        book=new int[N];
        for (int i = 0; i <N ; i++) {
            for (int j = 0; j <N ; j++) {
                e[i][j]=sc.nextInt();
            }
        }
        int count=0;
        for (int i = 0; i <N ; i++) {
            if(book[i]==0){
                count++;
                bfs(i);
            }
        }
        System.out.println(count);
    }
    /***
     * @函数功能：树的广度遍历
     * @param cur:当前结点
     * @return：void
     */
    public static void bfs(int cur){
        LinkedList queue=new LinkedList();
        queue.add(cur);
        while (!queue.isEmpty()){
            int temp=(int)queue.poll();
            book[temp]=1;
            for (int i = 0; i <N ; i++) {
                if(temp!=i&&e[temp][i]==1&&book[i]==0){
                    queue.offer(i);
                }
            }
        }

    }
}

 测试用例优先，各位读者也可以自行编写测试用例来调试此方法的可行性。