poj2761主席树

题意：求区间的第k小值

思路：主席树裸题，见我poj2104的分析

#include<cstdio>
#include<iostream> 
#include<algorithm>
#define N 100009
using namespace std;
int cnt;
struct node2
{
	int l,r,num; 
} s[20*N];
struct node1
{
	int id,num; 
} a[N];
int rank[N],root[N];
bool cmp(node1 a,node1 b)
{
	return a.num<b.num;
}
void insert(int num,int &x,int l,int r)
{
	s[cnt++]=s[x];
	x=cnt-1;
	s[x].num++;
	if(l==r) return;
	int mid=(l+r)/2; 
	if(num<=mid)
	insert(num,s[x].l,l,mid);
	else
	insert(num,s[x].r,mid+1,r);
}
int query(int u,int v,int w,int l,int r)
{
	if(l==r) return l;
	int t=s[s[v].l].num-s[s[u].l].num;
	int mid=(l+r)/2;
	if(t>=w)
	{
		return query(s[u].l,s[v].l,w,l,mid);
	}else
	{
		return query(s[u].r,s[v].r,w-t,mid+1,r);
	}
}
int main()
{
	//FILE *fp=fopen("t.txt","r");
	int n,m,u,v,w;
	root[0]=0;
	s[0].l=s[0].r=s[0].num=0;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i].num);
			a[i].id=i;
		}
		sort(a+1,a+1+n,cmp);
		for(int i=1;i<=n;i++)
		{
			rank[a[i].id]=i;
		}
		cnt=1;
		for(int i=1;i<=n;i++)
		{
			root[i]=root[i-1];//这就是让下一个树复制上一棵树 
			insert(rank[i],root[i],1,n);
		}
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			printf("%d\n",a[query(root[u-1],root[v],w,1,n)].num);			
		} 
	}
	return 0;
}