poj1988 Cube Stacking（关系并查集）

来源：互联网发布：关于网络暴力的数据编辑：程序博客网时间：2024/05/16 01:30

http://poj.org/problem?id=1988

题意：给你30000个箱子，M x y 代表把x所在的堆移动到y上，C x代表查询x号箱子下面的箱子数量。

思路：以每一堆最下面的箱子为根节点建树，below[N]代表某箱子下面的箱子数量，stackk[N]代表某一堆总共的箱子个数。Union操作只是将两堆根节点的相应参数变化，这样就合并成了一堆。箱子总数在递归过程中累加即可，重点是Find函数中below关系的处理。每次路径压缩将某个箱子的父节点改为根节点，其关系也要跟着变化。那当前的below值就等于父节点路径压缩前与当前节点的below值加上路径压缩后根节点与父节点的below值，每递归一次实现一次，递归真是神奇的东西啊！不过注意这里的Union操作只是改变根节点的值和参数，根节点上的参数并未改变，所以查询前要执行一次将值和参数更新。

#include <stdio.h>#include <algorithm>#include <stdlib.h>#include <string.h>#include <iostream>using namespace std;typedef long long LL;const int N = 100010;const int INF = 0x3f3f3f3f;int pre[N], below[N], stackk[N];int Find(int x){    if(x == pre[x]) return x;    int tmp = pre[x];    pre[x] = Find(pre[x]);    below[x] += below[tmp];    return pre[x];}void Union(int x, int y){    int rx = Find(x);    int ry = Find(y);    pre[rx] = ry;    below[rx] += stackk[ry];    stackk[ry] += stackk[rx];}int main(){   // freopen("in.txt", "r", stdin);    int p, x, y;    char s[5];    scanf("%d", &p);    for(int i = 1; i <= 30003; i++)    {        pre[i] = i;        below[i] = 0;        stackk[i] = 1;    }    while(p--)    {        scanf("%s", s);        if(s[0] == 'M')        {            scanf("%d%d", &x, &y);            Union(x, y);        }        else if(s[0] == 'C')        {            scanf("%d", &x);            Find(x);            printf("%d\n", below[x]);        }    }    return 0;}

0 0