poj2104 K-th Number（划分树）

划分树模板题：

给定1~n的一种排列，m次询问区间[i,j]中第k小的数 (1 <= n <= 100 000, 1 <= m <= 5 000). 

这篇文章对划分树讲得非常详细：http://blog.csdn.net/shiqi_614/article/details/8041390

细节见代码注释

#include<stdio.h>
#include<stdlib.h>
struct Node
{
	int v[100005],num[100005];//T[i].v[j]:划分树第i层的第j个位置的权值, T[i].num[j]:划分树第i层的第left~j个位置有多少个进了下一层的左孩子 
};
Node T[20];
int a[100005]={0};
void jh(int* a,int* b)
{
	int t=*a;
	*a=*b;
	*b=t;
}
void kp(int low,int high)
{
	int i=low,j=high,mid=a[(i+j)/2];
	while(i<j)
	{
		while(a[i]<mid) i++;
		while(a[j]>mid) j--;
		if(i<=j)
		{
			jh(&a[i],&a[j]);
			i++;
			j--;
		}
	}
	if(j>low) kp(low,j);
	if(i<high) kp(i,high);
}
void build(int left,int right,int deep)//求出这一层的num，和下一层的v
{
	int mid=(left+right)/2,ln=left,rn=mid+1,isame=mid-left+1,same=0,i;
	if(left==right) return;//没有下一层 
	for(i=left;i<=right;i++)
		if(T[deep].v[i]<a[mid]) isame--;//isame:下一层的左孩子中权值==a[mid]的有多少个 
	for(i=left;i<=right;i++)
	{
		if(i==left) T[deep].num[i]=0;//求num用到了前缀和的思想，这里把首位的num清零，相当于初始化 
		else T[deep].num[i]=T[deep].num[i-1];
		//接下来判断：该把第i个位置的数往下一层哪个孩子中放 
		if(T[deep].v[i]<a[mid])//小于中位数，放左边 
		{
			T[deep+1].v[ln++]=T[deep].v[i];
			T[deep].num[i]++;//计数器加1 
		}
		else
		{
			if(T[deep].v[i]>a[mid]) T[deep+1].v[rn++]=T[deep].v[i];//大于中位数，放右边 
			else//等于中位数，看左边还有没有位置放 
			{
				same++;
				if(isame>=same)//左边有位置 
				{
					T[deep+1].v[ln++]=T[deep].v[i];
					T[deep].num[i]++;
				}
				else  T[deep+1].v[rn++]=T[deep].v[i];
			}
		}
	}
	build(left,mid,deep+1);
	build(mid+1,right,deep+1);
}
int cx(int x,int y,int k,int left,int right,int deep)
{
	int mid=(left+right)/2,before,have,b,h;
	if(left==right) return T[deep].v[left];
	if(x==left)//特殊情况：区间左端点恰为左边界
	{
		before=0;
		have=T[deep].num[y];
	}
	else
	{
		before=T[deep].num[x-1];
		have=T[deep].num[y]-T[deep].num[x-1];
	}
	if(k<=have) return cx(left+before,left+before+have-1,k,left,mid,deep+1);//第k小的数在下一层左孩子中 
	else
	{
		b=x-1-left+1-before;//与before意思相反：left~x-1中属于下一层右孩子的点的个数
		h=y-x+1-have;//与have意思相反：x~y中属于下一层右孩子的点的个数
		return cx(mid+1+b,mid+1+b+h-1,k-have,mid+1,right,deep+1);//第k小的数在下一层右孩子中 
	}
}
int main()
{
	int n,m,i,j,k;
	while(scanf("%d%d",&n,&m)==2)
	{
		for(i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			T[0].v[i]=a[i];
		}
		kp(1,n);
		build(1,n,0);
		for(;m>0;m--)
		{
			scanf("%d%d%d",&i,&j,&k);
			printf("%d\n",cx(i,j,k,1,n,0));//与线段树一样，需要在递归时带上查询区间与边界 
		}
	}
	return 0;
}