BZOJ 2049 LCT

思路:LCT的基本操作

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 233333
int n,m,xx,yy,rev[N];
char op[11];
struct Tree{int l,r,where,num,fa;}t[N];
void reverse(int k){if(!k)return;rev[k]^=1;swap(t[k].l,t[k].r);}
void pushdown(int k){if(rev[k])reverse(t[k].l),reverse(t[k].r),rev[k]=0;}
void zig(int k){
    int f=t[k].fa;pushdown(f),pushdown(k);
    if(t[t[f].fa].l==f)t[t[f].fa].l=k;else if(t[t[f].fa].r==f)t[t[f].fa].r=k;
    t[k].fa=t[f].fa,t[f].l=t[k].r,t[t[f].l].fa=f,t[k].r=f;t[f].fa=k;}
void zag(int k){
    int f=t[k].fa;pushdown(f),pushdown(k);
    if(t[t[f].fa].l==f)t[t[f].fa].l=k;else if(t[t[f].fa].r==f)t[t[f].fa].r=k;
    t[k].fa=t[f].fa;t[f].r=t[k].l,t[t[f].r].fa=f;t[k].l=f;t[f].fa=k;}
int splay_root(int k){return t[k].fa&&(t[t[k].fa].l==k||t[t[k].fa].r==k);}
void splay(int k){pushdown(k);while(splay_root(k))if(t[t[k].fa].l==k)zig(k);else zag(k);}
int access(int k){int now=0;while(k)splay(k),t[k].r=now,now=k,k=t[k].fa;return now;}
void makeroot(int k){reverse(access(k)),splay(k);}
void link(int k1,int k2){makeroot(k1),t[k1].fa=k2;access(k1);}
void cut(int k1,int k2){makeroot(k1),access(k2),splay(k2),t[k2].l=t[k1].fa=0;}
int findfa(int k){while(t[k].fa)k=t[k].fa;return k;}
int find(int k1,int k2){makeroot(k1);if(findfa(k2)==k1)return 1;return 0;}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%s%d%d",op,&xx,&yy);
        if(op[0]=='Q'){
            if(find(xx,yy))puts("Yes");
            else puts("No");
        }
        else if(op[0]=='C')link(xx,yy);
        else cut(xx,yy);
    }
}

这里写图片描述

posted @ 2016-12-27 18:01  SiriusRen  阅读(...)  评论(... 编辑 收藏