poj2104 主席树

题目大意：给一串数字，多次询问区间的第k小值

思路：不带修改的主席树。第一次打主席树，所以还是参照了以下网上的模板。主席树的原理：

其实就是每添加一个数，就增加一棵树（其实是一条链），然后这条链上的计数值都+1,表示这个树链比之前的要多加了一个数。这样，想要比较第i个数与第j个数区间内，第k大的数就很好求了。只要比较s[i-1]和s[j]即可（即第i-1条树链，第j条树链），因为这两条树链的区间上计数的差值，都是[i,j]区间内的数一个个加上去的。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
struct node 
{
	int id,num;
}a[100009];
int cnt;
struct node2
{
	int l,r,num;
}s[100009*20];
int rank[100009],root[100009];
bool cmp(node a,node 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 a,int b,int c,int l,int r)
{
	if(l==r)
	return l;
	int t=s[s[b].l].num-s[s[a].l].num;
	int mid=(l+r)/2;
	if(c<=t)
		return query(s[a].l,s[b].l,c,l,mid);
	else
		return query(s[a].r,s[b].r,c-t,mid+1,r);
}
int main()
{
//	FILE *fp=fopen("t.txt","r");
	int n,m,a1,b1,c1;
	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;
	 root[0]=0;s[0].l=s[0].r=s[0].num=0;
	 for(int i=1;i<=n;i++)
	 {
	 	root[i]=root[i-1];
	 	insert(rank[i],root[i],1,n);
	 }
	 for(int j=1;j<=m;j++)
	 {
	 	scanf("%d%d%d",&a1,&b1,&c1);
	 	printf("%d\n",a[query(root[a1-1],root[b1],c1,1,n)].num);
	 }
	}
	return 0;
}