The Suspects————北大OJ

题目链接
http://poj.org/problem?id=1611
思路
传统并查集+维护一个存放每种个数的size数组,本题要找多少个感染的,即找与0有关系的,那么在读入每组数据的时候需要找到每组数据的最小值,以它为代表。
c++代码

#include<algorithm>
#include<iostream>
#include<stdio.h>
using namespace std;
const int N=30005;
int n,m;
int size[N];
int p[N];
int q[N];
int find(int x)
{
    
    
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x]; 
}
void init()
{
    
    
    for (int i = 0; i < n; i++)
    {
    
    
        p[i] = i;
        size[i] = 1;
    }
}
int main()
{
    
    
    cin >> n >> m;
    while(n||m)
    {
    
    
        if(n==0&&m==0)
        {
    
    
            break;
        }
        if(m==0)
        {
    
    
            printf("1\n");
        }
        else
        {
    
    
        init();
        while(m--)
        {
    
       
            int min1=N-1;
            int x;
            scanf("%d",&x);
            for (int k = 1; k <= x;k++)
            {
    
    
                cin >> q[k];
                min1 = min(min1, q[k]);
            }
            int b=find(min1);
            for (int k = 1; k <= x;k++)
            {
    
    
                if(q[k]==min1)
                    continue;
                int y = find(q[k]);
                if(y!=b)
                    p[y] = b, size[b] += size[y];
            }
        }
        cout << size[find(0)] << endl;
        }
        cin >> n >> m;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Star_Platinum14/article/details/111937637