BZOJ[4399]魔法少女LJJ 线段树合并

题目链接http://www.lydsy.com/JudgeOnline/problem.php?id=4399

c<=7！！c<=7！！c<=7！！

每新建一个节点，就开一个权值线段树，在连接时将两个线段树合并即可，对于操作3,4，直接查找比k大/小的数有多少个，查找时顺便将范围内的点删除，再全放在k上即可
操作五像平衡树那样随便搞一搞就可以了..
对于操作六，根据对数的运算性质$log(a*b)=log(a)+log(b)$，可以考虑在权值线段树上顺便维护一下区间内对数和，转换为区间加法

代码如下：

#include<ctype.h>
#include<cstdio>
#include<cmath>
#define N 400050
using namespace std;
const int INF=1000000000;
inline int read(){
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
int n,t,x,y,k,top;
int f[N];
int Find(int k){
    if(f[k]==k) return k;
    return f[k]=Find(f[k]);
}
struct Node{
    int l,r,sum;
    double sum1;
    Node *ls,*rs;
    Node(int,int);
    inline void maintain(){
        sum=ls->sum+rs->sum;
        sum1=ls->sum1+rs->sum1;
        return;
    }
}*root[N],*null;
Node::Node(int _,int __):l(_),r(__){
    ls=rs=null;
    sum1=sum=0;
}
void Merge(Node *&x,Node *&y){
    if(y==null) return;
    if(x==null){
        x=y;
        return;
    }
    x->sum+=y->sum;
    x->sum1+=y->sum1;
    Merge(x->ls,y->ls);Merge(x->rs,y->rs);
}
void Insert(int x,int v,Node *&k,int L,int R){
    if(k==null) k=new Node(L,R);
    if(k->l==k->r){
        k->sum+=v;
        k->sum1=k->sum*log(L);
        return;
    }
    int mid=L+R>>1;
    if(x<=mid) Insert(x,v,k->ls,L,mid);
    else Insert(x,v,k->rs,mid+1,R);
    k->maintain();
}
int Query(int x,int y,Node *&k){
    if(k==null) return 0;
    if(k->l>=x  && k->r<=y){
        int t=k->sum;
        k=null;
        return t;
    }
    int mid=k->l+k->r>>1,t;
    if(mid>=y) t=Query(x,y,k->ls);
    else if(mid<x) t=Query(x,y,k->rs);
    else t=Query(x,y,k->ls)+Query(x,y,k->rs);
    if(k->l>=x && k->r<=y) k=null;
    k->maintain();
    return t;
}
int K_th(int x,Node *k){
    if(k->l==k->r) return k->l;
    if(k->ls->sum>=x) return K_th(x,k->ls);
    return K_th(x-k->ls->sum,k->rs);///求第k大类似于平衡树
}
void Debug(Node *x){///调试orz....
    if(x->ls!=null) Debug(x->ls);
    if(x->rs!=null) Debug(x->rs);
    printf("%d %d %d  %lf\n",x->l,x->r,x->sum,x->sum1);
}
int main(){
    null=new Node(0,0);
    null->ls=null->rs=null;
    null->sum=null->sum1=0;
    n=read();
    for(int i=1;i<=n;i++){
        t=read();
        switch(t){
            case 1: x=read(); Insert(x,1,(root[++top]=new Node(1,INF)),1,INF); f[top]=top; break;
            case 2: x=Find(read()); y=Find(read()); if(x!=y) Merge(root[x],root[y]); f[y]=x; break;
            case 3: x=Find(read()); y=read(); k=Query(1,y-1,root[x]); Insert(y,k,root[x],1,INF); break;
            case 4: x=Find(read()); y=read(); k=Query(y+1,INF,root[x]); Insert(y,k,root[x],1,INF); break;
            case 5: x=Find(read()); y=read(); printf("%d\n",K_th(y,root[x])); break;
            case 6: x=Find(read()); y=Find(read()); puts(root[x]->sum1>root[y]->sum1?"1":"0"); break;
            case 7: x=Find(read()); printf("%d\n",root[x]->sum); break;
        }
    }
return 0;
}