PAT甲级1107 Social Clusters

最新推荐文章于 2024-09-14 22:25:23 发布

哈哈哈~哈哈哈

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量65

点赞数

文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_63953838/article/details/131023796

版权

该文阐述了一个关于社交网络的问题，通过每个人喜欢的活动来确定他们所在的社交网络。使用并查集数据结构，初始化每个人为独立集合，当发现两个人有相同的活动时，将他们所在的集合合并。通过查找根节点和集合合并操作，最后计算出社交网络的总数。代码示例展示了如何实现这一算法。

摘要由CSDN通过智能技术生成

1.题目描述

N个人，每个人喜欢若干项活动，若两个人中有任意一项活动相同，则称该两人处于同一个社交网络，若A和B在一个网络，B和C在一个网络，则称A和B和C在一个网络，求N个人形成了多少个社交网络。

2.思路

首先判断两个人是否处于同一个社交网络：定义数组course【i】，表示喜欢活动【i】的人的编号，初始化为0，findFather（course【i】）寻找course【i】所在社交网络的根节点。然后Union（i， findFather（course【i】）：将当前读入的人的编号和社交网络根节点合并为一个集合。其中isRoot【i】表示以i为根节点的社交网络的人数，ans 表示社交网络的的个数。father集合存放父亲节点。

3.参考代码

#include<cstdio>
#include<algorithm>
using namespace std;
const int N = 1010;
int father[N];
int isRoot[N] = {0};
int course[N] = {0};
int findFather(int x) {
    int a = x;
    while (x != father[x]) { 
        x = father[x];
    }
    while (a != father[a]) {
        int z= a;
        a = father[a];
        father[z] = x;
    }
    return x;
}
void Union(int a, int b){
    int faA = findFather(a);
    int faB = findFather(b);
    if (faA != faB)
        father[faA] = faB;
}
void init(int n){
    for (int i = 1; i <= n; i++){
        father[i] = i;
    }
}
bool cmp(int a, int b){
    return a > b;
}
int main(){
    int n, k, h;
    scanf("%d", &n);
    init(n);
    for (int i = 1; i <= n; i++) {
        scanf("%d:", &k);
        for (int j = 0; j < k; j++) {
            scanf("%d", &h);
            if (course[h] == 0)
                course[h] =i;
            Union(i, findFather(course[h]));
        }
    }
    int ans = 0;
    for (int i = 1; i <= n; i++) {
        isRoot[findFather(i)] ++;
    }
    for (int i = 1; i <= n; i++){
        if (isRoot[i] != 0)
            ans ++;
    }
    printf("%d\n", ans);
    sort(isRoot + 1, isRoot + n + 1, cmp);
    for (int i = 1; i <= ans; i++) {
        printf("%d", isRoot[i]);
        if (ans > i)
            printf(" ");
    }
    return 0;
}

4.感悟

该题考查的是并查集的模型，题目中的社交网络可以看作集合。