【POJ】【P2104&P2761】【题解】【区间K大】【主席树】

传送门：

http://poj.org/problem?id=2104

http://poj.org/problem?id=2761

裸主席树……

Code：

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
int a[maxn],b[maxn];
struct node{
	int size;
	node *c[2];
	node(){
		c[0]=c[1]=NULL;size=0;
	}
	void rz(){
		if(c[0])size+=c[0]->size;
		if(c[1])size+=c[1]->size;
	}	
}*Null=new node(),*root[maxn]={NULL};
int tot;
node poor[maxn<<4];
void insert(node *&y,node *&x,int l,int r,int t){
	if(x==NULL)x=Null;
	
	if(tot<(maxn<<4)){
		y=&poor[tot++];
		y->c[0]=y->c[1]=Null;
		y->size=0;
	}else
		y=new node();
	
	int mid=(l+r)>>1;
	if(l==r){
		*y=*x;
		y->size++;
		return;
	}
	if(t<=b[mid]){
		insert(y->c[0],x->c[0],l,mid,t);
		y->c[1]=x->c[1];
		y->rz();
	}else{
		insert(y->c[1],x->c[1],mid+1,r,t);
		y->c[0]=x->c[0];
		y->rz();
	}
}
int kth(node *&x,node *&y,int l,int r,int k){
	if(x==NULL)x=Null;
	if(y==NULL)y=Null;
	if(l==r)return b[l];
	int mid=(l+r)>>1;
	int R=0;
	if(y->c[0])R+=y->c[0]->size;
	if(x->c[0])R-=x->c[0]->size;
	if(R>=k)return kth(x->c[0],y->c[0],l,mid,k);
	else return kth(x->c[1],y->c[1],mid+1,r,k-R);
}
int n,m,size;
int main(){
	Null->c[0]=Null->c[1]=Null;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		b[i]=a[i];
	}
	sort(b+1,b+1+n);
	size=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++)
		insert(root[i],root[i-1],1,size,a[i]);
	while(m--){
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",kth(root[l-1],root[r],1,size,k));
	}
	return 0;
}

新模版：

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
inline int lowbit(int x){
	return x&(-x);
}
struct node{
	int size;
	node *c[2];
	node();
	void rz(){
		size=c[0]->size+c[1]->size;
	}
}*Null=new node(),*root[maxn];
node poor[maxn<<4];
int tot;
node::node(){
	size=0;
	c[0]=c[1]=Null;
}
int a[maxn],b[maxn];
void insert(node *&y,node *&x,int l,int r,int t){
	//if(x==NULL)x=Null;
	if(tot<(maxn<<4)){
		y=&poor[tot++];
		y->size=0;y->c[0]=y->c[1]=Null;
	}else	y=new node();
	if(l==r){
		*y=*x;y->size++;return;
	}
	int mid=(l+r)>>1;
	if(t<=b[mid]){
		insert(y->c[0],x->c[0],l,mid,t);
		y->c[1]=x->c[1];
		y->rz();
	}else{
		insert(y->c[1],x->c[1],mid+1,r,t);
		y->c[0]=x->c[0];
		y->rz();
	}
}
int kth(node *&x,node *&y,int l,int r,int k){
	if(l==r)return b[l];
	int mid=(l+r)>>1;
	int R=y->c[0]->size-x->c[0]->size;
	if(R>=k)return kth(x->c[0],y->c[0],l,mid,k);
	else return kth(x->c[1],y->c[1],mid+1,r,k-R);
}
int n,m,size;
int main(){
	Null->size=0;
	Null->c[0]=Null->c[1]=Null;
  	scanf("%d%d",&n,&m);
  	root[0]=Null;
	for(int i=1;i<=n;i++){
		root[i]=Null;
		scanf("%d",&a[i]);
		b[i]=a[i];
	}
	sort(b+1,b+1+n);
	size=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++)
		insert(root[i],root[i-1],1,size,a[i]);
	while(m--){
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",kth(root[l-1],root[r],1,size,k));
	}	
	return 0;
}