L3-003 社交集群(Python3)

最新推荐文章于 2024-08-23 16:53:34 发布

敲代码的小柯

最新推荐文章于 2024-08-23 16:53:34 发布

阅读量450

点赞数 1

本文链接：https://blog.csdn.net/weixin_55730361/article/details/128013346

版权

社交网络兴趣爱好并查集用户群组算法实现

关键词由CSDN通过智能技术生成

PTA 同时被 3 个专栏收录

117 篇文章 21 订阅

订阅专栏

天梯赛

106 篇文章 15 订阅

订阅专栏

python

96 篇文章 4 订阅

订阅专栏

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

输入格式：

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

Ki: hi[1] hi[2] ... hi[Ki]

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

输出格式：

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

输入样例：

输出样例：

3
4 3 1

提交结果：

代码:

def init(n):
    for i in range(1, n + 1):
        f[i] = i


def find(x):
    if x == f[x]:
        return x

    f[x] = find(f[x])
    return f[x]


def merge(x, y):
    a = find(x)
    b = find(y)
    if a != b:
        f[b] = a


n = int(input())
f = [0 for _ in range(1001)]
flag = [0 for _ in range(1001)]
num = [0 for _ in range(1001)]
init(n)
for i in range(1, n + 1):
    data = input().split()
    k = int(data[0][0])
    for id in data[1:]:
        id = int(id)
        if not flag[id]:
            flag[id] = i
        else:
            merge(i, flag[id])
for i in range(1, n + 1):
    x = find(i)
    num[x] += 1
cnt = 0
for i in range(1, n + 1):
    if num[i]:
        cnt += 1
print(cnt)
num = num[:n + 1]
num.sort(reverse=True)
print(num[0], end="")
for i in range(1, n + 1):
    if num[i]:
        print(" " + str(num[i]), end="")
    else:
        break