Kth number - hdu2664 - 归并树解法

思路：

（1）建树

先说一下merge函数：摘自此博客

merge函数的作用是：将两个有序的序列合并为一个有序的序列。函数参数：merge(first1,last1,first2,last2,result,compare);

//firs1t为第一个容器的首迭代器，last1为第一个容器的末迭代器，first2为第二个容器的首迭代器，last2为容器的末迭代器，result为存放结果的容器，comapre为比较函数（可略写，默认为合并为一个升序序列）。
建树过程：

相当于归并排序，自底向上归并的时候，用merge函数排一下序

void build(int l,int r,int rt){
    if(l==r){
        tree[rt].push_back(a[l]);
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    tree[rt].resize(r-l+1);
    merge(tree[rt<<1].begin(),tree[rt<<1].end(),tree[rt<<1|1].begin(),tree[rt<<1|1].end(),tree[rt].begin());
}

（2）查询

是二分套二分的思想，第一次二分是在排好序的数组里二分

选择一个数字，然后进行第二次二分查找，看它在当前区间的排名，

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
const int N=100005;
int n,m;
int a[N],tmp[N];
vector<int>tree[N<<2];

void build(int l,int r,int rt){
    if(l==r){
        tree[rt].push_back(a[l]);
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    tree[rt].resize(r-l+1);
    merge(tree[rt<<1].begin(),tree[rt<<1].end(),tree[rt<<1|1].begin(),tree[rt<<1|1].end(),tree[rt].begin());
}

int query(int L,int R,int x,int l,int r,int rt){//查找元素x在区间[L,R]是第几大
    if(L<=l&&R>=r){
        return upper_bound(tree[rt].begin(),tree[rt].end(),x)-tree[rt].begin();
    }
    int m=(l+r)>>1;
    int sum=0;
    if(L<=m)sum+=query(L,R,x,lson);
    if(R>m)sum+=query(L,R,x,rson);
    return sum;
}

int solve(int L,int R,int k,int n){//二分查找tmp[m]这个元素是不是第k大
    int l=1,r=n,m;
    while(l+1<r){
        m=(l+r)>>1;
        if(query(L,R,tmp[m],1,n,1)>=k){
            r=m;
        }
        else l=m;
    }
    return tmp[r];
}

int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        memset(tree,0,sizeof(tree));
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            tmp[i]=a[i];
        }
        build(1,n,1);
        sort(tmp+1,tmp+1+n);
        while(m--){
            int l,r,k;
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",solve(l,r,k,n));
        }
    }
}