poj 2104 K-th Number

题目链接：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;
}