PTA-Python-朋友圈

PTA Python 朋友圈

某学校有N个学生，形成M个俱乐部。每个俱乐部里的学生有着一定相似的兴趣爱好，形成一个朋友圈。一个学生可以同时属于若干个不同的俱乐部。根据“我的朋友的朋友也是我的朋友”这个推论可以得出，如果A和B是朋友，且B和C是朋友，则A和C也是朋友。请编写程序计算最大朋友圈中有多少人。

输入格式：
输入的第一行包含两个正整数N（≤30000）和M（≤1000），分别代表学校的学生总数和俱乐部的个数。后面的M行每行按以下格式给出1个俱乐部的信息，其中学生从1~N编号：
第i个俱乐部的人数Mi（空格）学生1（空格）学生2 … 学生Mi
输出格式：
输出给出一个整数，表示在最大朋友圈中有多少人。
输入样例：

7 4
3 1 2 3
2 1 4
3 5 6 7
1 6

输出样例：
4

代码如下：

fa=[0 for i in range(30005)]
cn=[0 for i in range(30005)]
t = 0
def find(x):
    if x == fa[x]:
        return x
    else:
        fa[x] = find(fa[x])
        return fa[x]


n,m = map(int,input().split())
for  i in range(1,n+1):
    fa[i] = i
    cn[i] = 1
while m != 0:
    m -= 1
    A = input()
    B = [int(n) for n in A.split()]
    for i in range(1,len(B)-1):
        x = find(B[i])
        y = find(B[i+1])
        if x != y:
            fa[x] = y
            cn[y] += cn[x]
        B[i] = B[i+1]
for i in range(n+1):
    t = max(t,cn[i])
print(t)

求赞