【并查集-size[ ]】POJ 1611 The Suspects

最新推荐文章于 2022-03-03 22:06:14 发布

原创最新推荐文章于 2022-03-03 22:06:14 发布 · 137 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

数据结构

并查集

本文深入解析了POJ1611题目《The Suspects》的解决方案，通过并查集算法确定嫌疑人群体。文章阐述了如何将学生群体进行合并，最终计算与初始嫌疑学生相连的总人数。

POJ 1611 The Suspects

题意：

n个学生，序号从0到n-1。m组group，如果这个小组有一个是suspect，那么整组都是suspect。现规定0是suspect，问所有学生里有多少个suspect。

思路：

我们将每个组的小组成员Merge起来。当然这时候我们并查集里的数组size[ ]（用于表示该节点下一个有多少个子节点，包括自身）就要用到了，最后我们需要的答案是和0在一颗树上的节点数，也就是size[ Find(0) ]。
End

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <list>
#include <map>
#define INF 0x3f3f3f3f
#define MID l + r >> 1
#define lsn rt << 1
#define rsn rt << 1 | 1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr

using namespace std;
typedef long long ll;
const int maxN = 3e4 + 5;
int n, m, k, num[maxN];

int root[maxN], high[maxN], size[maxN];
void init()
{
    for(int i = 0 ; i < n ; i ++)
    {
        root[i] = i;
        high[i] = 0;
        size[i] = 1;
    }
}
int Find(int x) { return root[x] == x ? x : root[x] = Find(root[x]); }
int Same(int x, int y ){ return Find(x) == Find(y); }
void Merge(int x, int y)
{
    x = Find(x);
    y = Find(y);
    if(high[x] < high[y] || size[x] < size[y])
    {
        root[x] = y;
        size[y] += size[x];
    }
    else
    {
        root[y] = x;
        size[x] += size[y];
        if(high[x] == high[y]) high[x] ++ ;
    }
}
void f()
{
    for(int i = 1; i < k; i ++ )
    {
        if(!Same(num[i], num[i - 1]))
            Merge(num[i], num[i - 1]);
    }
 }

int main()
{
    while(~scanf("%d%d", &n, &m) && (n || m))
    {
        init();
        while(m -- )
        {
            scanf("%d", &k);
            for(int i = 0 ; i < k ; i ++ )
                scanf("%d", &num[i]);
            f();
        }
        printf("%d\n", size[Find(0)]);
    }
    return 0;
}