poj 1611 The Suspects

本文介绍两种算法实现疫情传播模拟:一种使用标记感染个人和组别的方式进行递归感染,另一种运用并查集处理群体间感染关系,高效计算最终感染人数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

#include<stdio.h>///模拟做法
#include<string.h>
#include<iostream>
#include<vector>
using namespace std;
const int N=3e4+500;
const int M=5e2+500;
int vis[N];///标记被感染的人
int vispos[N];///标记被感染的组
vector<int>a[M];///第i组有哪些人
vector<int>d[N];///i是第几组的
void ganran(int b)///将第b组的人标记感染
{
    if(vispos[b]==1)
        return ;
    vispos[b]=1;///记录这一组已经被感染
    for(int j=0;j<a[b].size();j++)
    {
        vis[a[b][j]]=1;
        for(int k=0;k<d[a[b][j]].size();k++)
            ganran(d[a[b][j]][k]);
    }
}
int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        if(n==0&&m==0)
            break;
        ///init()初始化
        {
            memset(vis,0,sizeof(vis));
            memset(vispos,0,sizeof(vispos));
            vector<int>a1[M];
            vector<int>d1[N];
            swap(a,a1);
            swap(d,d1);
        }
        int num,id;
        vis[0]=1;
        for(int i=0;i<m;i++)
        {
            cin>>num;
            for(int j=0;j<num;j++)
            {
                cin>>id;
                a[i].push_back(id);
                d[id].push_back(i);
            }
        }
        for(int i=0;i<d[0].size();i++)///从有0的组开始感染
            ganran(d[0][i]);
        int ans=0;
        for(int j=0;j<n;j++)
        {
            if(vis[j]==1)
                ans++;
        }
        cout<<ans<<endl;
    }
    return 0;
}
#include<stdio.h>///并查集做法
#include<string.h>
#include<iostream>
#include<vector>
using namespace std;
const int N=3e4+500;
const int M=5e2+500;
int a[N],size[N];
int n,m;
void init()
{
    for(int i=0;i<n;i++)
        a[i]=i,size[i]=1;
}
int getf(int x)
{
    if(a[x]==x)
        return x;
    return a[x]=getf(a[x]);
}
void mem(int x,int y)
{
    int xx=getf(x);
    int yy=getf(y);
    if(xx!=yy)
    {
        if(xx==0)
            a[yy]=xx,size[xx]+=size[yy];
        else
            a[xx]=yy,size[yy]+=size[xx];
    }
    else
        return ;
}
int main()
{
    while(cin>>n>>m)
    {
        if(n==0&&m==0)
            break;
        init();
        int num,id;
        for(int i=0;i<m;i++)
        {
            cin>>num;
            int boss;
            for(int j=0;j<num;j++)
            {
                cin>>id;
                if(j==0)
                    boss=id;
                else
                    mem(boss,id);
            }
        }
        cout<<size[0]<<endl;
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值