蓝桥杯之--并查集练习

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

RedCap.

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

阅读量373

点赞数 5

分类专栏：蓝桥杯之我试文章标签：算法

本文链接：https://blog.csdn.net/wrj1472583690/article/details/136607371

版权

蓝桥杯之我试专栏收录该内容

20 篇文章 0 订阅

订阅专栏

文章描述了在处理一个社交关系场景中，如何用并查集数据结构来管理和合并用户的爱好，最终统计每个爱好拥有者数量。作者对传统二维数组方法进行了优化，采用了简化的一维结构进行存储和操作。

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

首先写下模版

def root(x):
    if(p[x]!=x):
        p[x]=root(p[x])
    return p[x]
def union(x,y):
    if(root(x)!=root(y)):
        p[root(x)]=root(y)

然后在分析，发现有一点不太一样。。。

好像不太能用并查集，但是

它跟社交关系网很相似，最多1000个爱好，我们开始想到一个【1000】【】的二维数组，每个爱好里面记录下他的主人，后来因为太麻烦，又有可能爆空间。

于是，我们想到了：只取一维的，每个里面只放入一个元素即可

每一次，和那一个元素union就行了

def root(x):
    if(p[x]!=x):
        p[x]=root(p[x])
    return p[x]
def union(x,y):
    if(root(x)!=root(y)):
        p[root(x)]=root(y)
n=int(input())
p=[i for i in range(n+1)]
h=[0 for i in range(1001)]
for i in range(1,n+1):
    b=list(map(int,input().split()[1:]))
    for j in b:
        if((h[j])==0):
            h[j]=i
        elif (i)!=h[j]:
            union(i,h[j])
            #print(i,h[j])
ss=[0 for i in range(1+n)]
ans=0
for ii in range(1,n+1):
    ss[root(ii)]+=1 
    if(ss[root(ii)]==1):ans+=1
print(ans)
ss.sort(reverse=True)
for i in ss:
    if(i!=0):print(i,end=' ')