L2-024. 部落(并查集)

并查集,背模板就AC了,基本模板就是getf函数merge函数,注意getf函数里面的que[k]=getf(que[k]),其中的que[k]=一定不能省略,这一步是压缩路径,如果少了这一步,非常容易运行超时!!!

#include<iostream>
#include<algorithm>
using namespace std;
int que[10005];
int getf(int k)
{
    return que[k]==k?k:que[k]=getf(que[k]);
}
/* int getf(int k)//运行超时 { return que[k]==k?k:getf(que[k]); } */ 
int merge(int a,int b)
{
    if (getf(a)!=getf(b))
    que[getf(a)]=getf(b); 
}
int main()
{
    int n,t,i,j,k,a,b,book[10005],count1=0,count2=0;
    fill(book,book+10005,0);
    for (i=0;i<10005;i++)
    que[i]=i;
    cin>>n;
    for (i=0;i<n;i++)
    {
        cin>>t;
        cin>>a;
        book[a]=1;
        for (j=1;j<t;j++)
        {
            cin>>b;
            book[b]=1;
            merge(a,b);
        }
    }
    for (i=0;i<10005;i++)
    {
        if (book[i]==1)
        {
            count1++;
            if (que[i]==i)
            count2++;
        }
    }
    cout<<count1<<" "<<count2<<endl;
    cin>>k;
    for (i=0;i<k;i++)
    {
        cin>>a>>b;
        if (getf(a)==getf(b))
        cout<<"Y"<<endl;
        else
        cout<<"N"<<endl;
    }
}
全部评论

相关推荐

老树开花:可以开始投了,不用等到觉得完全准备好。一边投一边根据面试反馈改简历是最高效的方式。简历上项目描述建议突出你解决的具体问题,比如编辑器的性能优化、大文档渲染怎么处理的,而不只是列技术栈。中厂前端实习现在竞争确实激烈,建议同时关注一些有AI业务的团队,前端加AI应用是很有差异化的组合。Vue全家桶基础扎实的话可以往SSR或者跨端方向延伸,这些是面试加分项。加油,时间还来得及。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务