划分树模板题:
给定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;
}