L3-003 社交集群 (30 分)-JAVA

当你在社交网络平台注册时，一般总是被要求填写你的个人兴趣爱好，以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式：

输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：

K_i: h_i[1] h_i[2] … h_i[Ki]
其中K_i (>0)是兴趣爱好的个数，h_i[j]是第j个兴趣爱好的编号，为区间 [1, 1000] 内的整数。

输出格式：

首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔，行末不得有多余空格。

输入样例：

8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4

输出样例：

3
4 3 1

并查集。。。。。按套路模板带进行了。。。。

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    static int n;  // 人数
    static int[] par = new int[1005];  // 用于并查集
    static int[] arr = new int[1005];  // 记录某人的一个爱好
    static int[] count = new int[1005];  // 每种根集所有人数的数量

    public static void main(String[] args) {

        Scanner scan = new Scanner(System.in);

        n = scan.nextInt();

        scan.nextLine();
        
        // 初始化par数组
        for (int i = 1; i <= 1000; i++) {
            par[i] = i;
        }

        for (int i = 1; i <= n; i++) {

            String[] ss = scan.nextLine().replace(":","").split(" ");
            // 提取出此人有几个爱好
            int m = Integer.parseInt(ss[0]);
            
            // 记录某人的一个兴趣爱好
            arr[i] = Integer.parseInt(ss[1]);
            
            // 合并
            for (int j = 1; j <= m - 1; j++) {
                int x = Integer.parseInt(ss[j]);
                int y = Integer.parseInt(ss[j+1]);
                merge(x,y);
            }
        }
        
        // 查找每个人属于哪个群体,将其统计到这个群体的人数里面
        for (int i = 1; i <= n; i++) {
            int p = findP(arr[i]);
            count[p]++;
        }

        // 对每个集群含有的人数排序
        Arrays.sort(count);
        
        // 统计有几个集群
        int c = 0;
        for (int i = 1; i < 1005; i++) {
            if (count[i]!=0) c++;
        }

        System.out.println(c);
        
        // 逆序输出每个集群的人数
        System.out.print(count[count.length-1]);
        for (int i = 2; i <= c; i++) {
            System.out.print(" " + count[count.length-i]);
        }

    }


    // 合并
    public static void merge(int x, int y){

        x = findP(x);

        par[x] = y;
    }

    // 查找子树的根
    public static int findP(int x){

        while (x!=par[x]){
            x = par[x];
        }

        return x;
    }
}