題目思路:劃分樹求區間第k大數。
[cpp]
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define M 110000
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
int tr[20][M],s[20][M];
int sorted[M];
void build(int d,int l,int r)
{
if(l==r) return ;
int i,j,k,mid;
mid=(l+r)>>1,j=l,k=mid+1;
for(i=l;i<=r;i++)
{
s[d][i]=s[d][i-1];
if(tr[d][i]<=sorted[mid])
{
s[d][i]++;
tr[d+1][j++]=tr[d][i];
}
else
{
tr[d+1][k++]=tr[d][i];
}
}
build(d+1,l,mid);
build(d+1,mid+1,r);
}
int getkth(int d,int lp,int rp,int l,int r,int k)
{
int lt,rt;
if(lp==rp) return tr[d][lp];
int mid=(lp+rp)>>1;
if(k<=s[d][r]-s[d][l-1])
{
lt=lp+s[d][l-1]-s[d][lp-1];
rt=lp+s[d][r]-s[d][lp-1]-1;
return getkth(d+1,lp,mid,lt,rt,k);
}
else
{
lt=mid+1+(l-lp)-(s[d][l-1]-s[d][lp-1]);
rt=mid+(r-lp+1)-(s[d][r]-s[d][lp-1]);
return getkth(d+1,mid+1,rp,lt,rt,k-(s[d][r]-s[d][l-1]));
}
}
int main()
{
int i,n,m,l,r,k;
while(scanf("%d%d",&n,&m)!=EOF)
{
for(i=1;i<=n;i++)
{
scanf("%d",&tr[0][i]);
sorted[i]=tr[0][i];
}
sort(sorted+1,sorted+n+1);
for(i=0;i<20;i++)
s[i][0]=0;
build(0,1,n);
// for(i=1;i<=n;i++)
// {
// printf("i %d s %d\n",i,s[0][i]);
// }
while(m--)
{ www.2cto.com
scanf("%d%d%d",&l,&r,&k);
printf("%d\n",getkth(0,1,n,l,r,k));
}
}
}
作者:Wings_of_Liberty