【线段树合并】【bzoj4399】: 魔法少女LJJ

Description

在森林中见过会动的树，在沙漠中见过会动的仙人掌过后，魔法少女LJJ已经觉得自己见过世界上的所有稀奇古怪的事情了
LJJ感叹道“这里真是个迷人的绿色世界,空气清新、淡雅,到处散发着醉人的奶浆味；小猴在枝头悠来荡去,好不自在；各式各样的鲜花争相开放,各种树枝的枝头挂满沉甸甸的野果；鸟儿的歌声婉转动听,小河里飘着落下的花瓣真是人间仙境”
SHY觉得LJJ还是太naive，一天，SHY带着自己心爱的图找到LJJ，对LJJ说：“既然你已经见识过动态树，动态仙人掌了，那么今天就来见识一下动态图吧”
LJJ：“要支持什么操作？”
SHY：“
1.新建一个节点，权值为x。
2.连接两个节点。
3.将一个节点a所属于的联通快内权值小于x的所有节点权值变成x。
4.将一个节点a所属于的联通快内权值大于x的所有节点权值变成x。
5.询问一个节点a所属于的联通块内的第k小的权值是多少。
6.询问一个节点a所属联通快内所有节点权值之积与另一个节点b所属联通快内所有节点权值之积的大小。
7.询问a所在联通快内节点的数量
8.若两个节点a，b直接相连，将这条边断开。
9.若节点a存在，将这个点删去。
”
LJJ：“我可以离线吗？”
SHY：“可以，每次操作是不加密的，”
LJJ：“我可以暴力吗？”
SHY：“自重”
LJJ很郁闷，你能帮帮他吗

Input

第一行有一个正整数m，表示操作个数。
接下来m行，每行先给出1个正整数c。
若c=1，之后一个正整数x，表示新建一个权值为x的节点，并且节点编号为n+1（当前有n个节点）。
若c=2，之后两个正整数a，b，表示在a，b之间连接一条边。
若c=3，之后两个正整数a，x，表示a联通快内原本权值小于x的节点全部变成x。
若c=4，之后两个正整数a，x，表示a联通快内原本权值大于x的节点全部变成x。
若c=5，之后两个正整数a，k，表示询问a所属于的联通块内的第k小的权值是多少。
若c=6，之后两个正整数a，b，表示询问a所属联通快内所有节点权值之积与b所属联通快内所有节点权值之积的大小，
若a所属联通快内所有节点权值之积大于b所属联通快内所有节点权值之积，输出1，否则为0。
若c=7，之后一个正整数a，表示询问a所在联通块大小
若c=8，之后两个正整数a，b，表示断开a，b所连接的边。
若c=9，之后一个正整数a，表示断开a点的所有连边
具体输出格式见样例

Sample Input

12
1 2
1 3
1 4
1 5
1 6
2 1 2
2 2 3
2 3 4
2 4 5
9 1
3 2 5
5 3 4

Sample Output

6

HINT

对100%的数据 0<=m<=400000,c<=7,所有出现的数均<=1000000000,所有出现的点保证存在

思路

这题吧，其实你看题面是会不会被c=8||9所吓倒
这题你只要认真读题面就可做

代码

#include <bits/stdc++.h>
#define N 100050
#define INF 1000000000
using namespace std;
inline int read(){
    int ret=0,f=1;char c=getchar();
    for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
    for(;isdigit(c);c=getchar())ret=ret*10+c-'0';
    return ret*f;
}
int m,n,fa[N];
int dfs(int x){return fa[x]==x?x:fa[x]=dfs(fa[x]);}
struct pppp{
    pppp *ch[2];
    int siz;double sum;
    pppp(){ch[0]=ch[1]=NULL;siz=0;sum=0;}
    void Pushdown(){
        siz=0;sum=0;
        if(ch[0])siz+=ch[0]->siz,sum+=ch[0]->sum;
        if(ch[1])siz+=ch[1]->siz,sum+=ch[1]->sum;
    }
}*root[N];
void Insert(pppp* &o,int pos,int v,int l,int r){
    if(!o)o=new pppp();
    if(l==r){
        o->siz+=v;
        o->sum=o->siz*log(l);
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)Insert(o->ch[0],pos,v,l,mid);
    else Insert(o->ch[1],pos,v,mid+1,r);
    o->Pushdown();
}
void Merge(pppp* &x,pppp* &y){
    if(!y)return ;
    if(!x){x=y;return ;}
    x->siz+=y->siz;x->sum+=y->sum;
    Merge(x->ch[0],y->ch[0]);
    Merge(x->ch[1],y->ch[1]);
}
void Clear(pppp* &o,int l,int r,int L,int R){
    if(!o)return ;
    if(l>r)return ;
    if(l==L&&r==R){
        delete(o);
        o=NULL;
        return ;
    }
    int mid=(L+R)>>1;
    if(r<=mid)Clear(o->ch[0],l,r,L,mid);
    else if(l>mid)Clear(o->ch[1],l,r,mid+1,R);
    else Clear(o->ch[0],l,mid,L,mid),Clear(o->ch[1],mid+1,r,mid+1,R);
    o->Pushdown();
}
int Q_siz(pppp* o,int l,int r,int L,int R){
    if(l>r||!o)return 0;
    if(l==L&&r==R)return o->siz;
    int mid=(L+R)>>1;
    if(r<=mid)return Q_siz(o->ch[0],l,r,L,mid);
    else if(l>mid)return Q_siz(o->ch[1],l,r,mid+1,R);
    else return Q_siz(o->ch[0],l,mid,L,mid)+Q_siz(o->ch[1],mid+1,r,mid+1,R);
}
int Q_kth(pppp* o,int x,int L,int R){
    if(L==R)return L;
    if(!o||o->siz<x)return -1;
    int mid=(L+R)>>1;
    int Lsiz=o->ch[0]?o->ch[0]->siz:0;
    if(Lsiz>=x)return Q_kth(o->ch[0],x,L,mid);
    else return Q_kth(o->ch[1],x-Lsiz,mid+1,R);
}
int main(){int opt,x,y;
    m=read();
    while(m--){
        opt=read();x=read();
        if(opt==1){
            Insert(root[++n],x,1,1,INF);
            fa[n]=n;
        }
        else if(opt==2){
            y=read();
            x=dfs(x);y=dfs(y);
            if(x==y)continue;
            fa[y]=x;
            Merge(root[x],root[y]);
        }
        else if(opt==3){
            y=read();x=dfs(x);
            int p=Q_siz(root[x],1,y-1,1,INF);
            Insert(root[x],y,p,1,INF);
            Clear(root[x],1,y-1,1,INF);
        }
        else if(opt==4){
            y=read();x=dfs(x);
            int p=Q_siz(root[x],y+1,INF,1,INF);
            Insert(root[x],y,p,1,INF);
            Clear(root[x],y+1,INF,1,INF);
        }
        else if(opt==5){
            y=read();x=dfs(x);
            printf("%d\n",Q_kth(root[x],y,1,INF));
        }
        else if(opt==6){
            y=read();
            x=dfs(x);y=dfs(y);
            printf("%d\n",root[x]->sum>root[y]->sum);
        }
        else{
            x=dfs(x);
            printf("%d\n",root[x]->siz);
        }
    }
    return 0;
}