题目链接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;
}