当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:
Ki: hi[1] hi[2] ... hi[Ki]
其中Ki(>0)是兴趣爱好的个数,hi[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
提交结果:
代码:
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