题目链接:poj 2104 K-th Number
题意:给你n个数,数在[1,n],中,即为[1,n]的不同排列方式,现在给你一个子区间,让你求这个区间第k小的数
归并树法:(复杂度o(n*log(n)+m*log^3n))
所谓归并树,就是利用归并排序的思想,将两个有序的数列合并为一个新的有序数列。
我们先提出一个概念,就是要求第k小的数x,则不超过x的数不少于k个,比如1,2,2,3,x=2,k=2,不超过x的有三个,第二,比x小的数不足k个,比如1,2,2,2, x=2,k=3,比x小的数只有一个,则求出不超过x的个数,就能算出x排第几,这里采用二分查找。
这里用线段树维护的不是一个点,而是一个序列,我们用动态数组来保存这个有序数列。
#include<iostream>
#include<vector>
#include<cstdio>
#include<algorithm>
#define maxn 111111
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
vector<int>x[maxn<<2];
int L[maxn],R[maxn],k[maxn],a[maxn],n,m;
void pushup(int rt)
{
merge(x[rt<<1].begin(),x[rt<<1].end(),x[rt<<1|1].begin(),x[rt<<1|1].end(),x[rt].begin());//将两子节点的序列排序到父节点
}
void build(int l,int r,int rt)
{
x[rt].clear(); //初始化
if(l==r)
{
x[rt].push_back(a[l-1]); //把区间的数一个一个写入叶子节点的动态数组
return ;
}
int m =(l+r)>>1;
build(lson);
build(rson);
x[rt].resize(r-l+1);//重载当前节点的长度,因为要合并左右儿子
pushup(rt);//向上进行归并排序
}
int query(int L,int R,int p,int l,int r,int rt)
{
if(R<l||L>r) return 0;//要查找的区间不在节点的左右区间时,直接返回0
if(L<=l&&r<=R) return upper_bound(x[rt].begin(),x[rt].end(),p)-x[rt].begin();//在区间内,则返回小于等于p的个数,upper_bound返回第一个大于p的地址
int m=(l+r)>>1;
int sum =0 ;//初值为0
if(L<=m) sum+=query(L,R,p,lson); //左右查询
if(R>m) sum+=query(L,R,p,rson);
return sum;
}
void binsearch()
{
sort(a,a+n);//先排序,因为后面要进行二分查找
for(int i=0;i<m;i++)
{
int left =0,right = n-1,mid;
while(left<right) //二分查找
{
mid =(left+right)>>1;
int p = a[mid];
int c = query(L[i],R[i],p,1,n,1);
if(c<k[i]) left=mid+1;//c为比当前查找点小或者等于的个数,若这个数目小于题中所求的第k小,则在右边继续找
else right=mid;//否则在左边查找
}
printf("%d\n",a[left]);
}
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=0;i<n;i++) scanf("%d",&a[i]); //先将区间的值保存在一数组
build(1,n,1);//建树
for(int i=0;i<m;i++) scanf("%d%d%d",&L[i],&R[i],&k[i]); //将所有查询保存在数组
binsearch();
}
return 0;
}