【bzoj1455】【罗马游戏】左偏树+并查集(模板)

这里写图片描述

Description
罗马皇帝很喜欢玩杀人游戏。 他的军队里面有n个人,每个人都是一个独立的团。最近举行了一次平面几何测试,每个人都得到了一个分数。 皇帝很喜欢平面几何,他对那些得分很低的人嗤之以鼻。他决定玩这样一个游戏。 它可以发两种命令: 1. Merger(i, j)。把i所在的团和j所在的团合并成一个团。如果i, j有一个人是死人,那么就忽略该命令。 2. Kill(i)。把i所在的团里面得分最低的人杀死。如果i这个人已经死了,这条命令就忽略。 皇帝希望他每发布一条kill命令,下面的将军就把被杀的人的分数报上来。(如果这条命令被忽略,那么就报0分)
Input
第一行一个整数n(1<=n<=1000000)。n表示士兵数,m表示总命令数。 第二行n个整数,其中第i个数表示编号为i的士兵的分数。(分数都是[0..10000]之间的整数) 第三行一个整数m(1<=m<=100000) 第3+i行描述第i条命令。命令为如下两种形式: 1. M i j 2. K i
Output
如果命令是Kill,对应的请输出被杀人的分数。(如果这个人不存在,就输出0)
Sample Input
5
100 90 66 99 10
7
M 1 5
K 1
K 1
M 2 3
M 3 4
K 5
K 4
Sample Output
10
100
0
66

很裸的一道左偏树的题,本来就是做来复习一下左偏树的。
然而却RE了。。。
左偏树通常都要和并查集连用,如何连用也是一个需要模板化的东西,我就在这上面挂了。
合并堆时,并查集这样改:

fa[fax]=fa[fay]=merge(fax,fay)

弹出堆顶元素时,并查集这样改才能避免RE:

fa[fax]=merge(le,ri);fa[fa[fax]]=fa[fax]

因为这个RE了好久,原来是因为在弹出堆顶元素之前,树中的元素的fa已经指向该对顶元素了,所以要将该元素的fa指向新生成的堆顶

完整代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=2000000+5;

struct node{
    int ls,rs;
    int dis,key;
}t[N];
int n,m,fa[N];
bool kill[N];

int getfa(int x){
    if(fa[x]==x) return x;
    return fa[x]=getfa(fa[x]);
}
int merge(int x,int y){
    if(x==0) return y;
    if(y==0) return x;
    if(t[x].key>t[y].key) swap(x,y);
    t[x].rs=merge(t[x].rs,y);
    if(t[t[x].ls].dis<t[t[x].rs].dis) swap(t[x].ls,t[x].rs);
    t[x].dis=t[t[x].rs].dis+1;
    return x;
}
int main(){
    t[0].ls=t[0].rs=0;
    t[0].dis=-1,t[0].key=0x3f3f3f3f;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&t[i].key);
        fa[i]=i;
    }
    scanf("%d",&m);
    char opt[2];
    int x,y;
    while(m--){
        scanf("%s",opt);
        if(opt[0]=='M'){
            scanf("%d%d",&x,&y);
            if(kill[x]||kill[y]) continue;
            int fax=getfa(x),fay=getfa(y);
            if(fax==fay) continue;
            int tmp=merge(fax,fay);
            fa[fay]=fa[fax]=tmp;
        }
        else{
            scanf("%d",&x);
            if(kill[x]){
                printf("0\n");continue;
            }
            int fax=getfa(x);
            printf("%d\n",t[fax].key);
            kill[fax]=1;
            int le=t[fax].ls,ri=t[fax].rs;
            fa[fax]=merge(le,ri);
            fa[fa[fax]]=fa[fax];
        }
    }
}
posted @ 2017-10-31 19:09  LinnBlanc  阅读(153)  评论(0编辑  收藏  举报