POJ 2104 归并树 OR 划分树

以前做的线段树都是存了一个值，而现在这颗树存的是一个数组。保存从[ l , r )这个左闭右开区间的有序值。

建树的方法就是归并排序的过程，因此叫做归并树。

如何查询某区间内的第K大值呢？

用到的一个技巧是2分答案。

我觉得2分是个很好用又很有技巧性的东西。首先是二分区间的选择，这里选择是右闭左开，因为当符合条件的优先更新右边。

对于第K大的数，不大于第K大的数的数量一定要大于等于K，所以对于这个数X，x越大越能满足这个条件，所以在满足条件的情况下不断的减小这个X的值，也就是更新右区间。

另外，STL有merge函数，还有个inplace_merge，还可以自己写compare函数。

#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;

vector<int> dat[500000];
//  STL 都是使用的左闭右开区间，自己的线段树最好也使用这样的区间
int A[100003];
int L[100003];
int R[100003];
int K[100003];
int nums[100003];

int build(int k,int l,int r)
{
    if(r-l==1)
    {
        dat[k].push_back(A[l]);
    }
    else
    {
        int mid=(l+r)>>1;
        int ls=k*2;
        int rs=k*2+1;
        build(ls,l,mid);
        build(rs,mid,r);
        dat[k].resize(r-l);
        merge(dat[ls].begin(),dat[ls].end(),dat[rs].begin(),dat[rs].end(),dat[k].begin());
    }
}
//不大于x的有多少个
int query(int a,int b,int k,int l,int r,int x)
{
    if(b<=l||a>=r)
        return 0;
    if(a<=l&&r<=b)
        return upper_bound(dat[k].begin(),dat[k].end(),x)-dat[k].begin();
    int mid=(l+r)>>1;
    int v1=query(a,b,k*2,l,mid,x);
    int v2=query(a,b,k*2+1,mid,r,x);
    return v1+v2;
}

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&A[i]);
        nums[i]=A[i];
    }
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&L[i],&R[i],&K[i]);
    sort(nums+1,nums+n+1);
    build(1,1,n+1);
    for(int i=1;i<=m;i++)
    {
        int ub=n;
        int lb=0;
        int k=K[i];
        while(ub-lb>1)
        {
            int mid=(ub+lb)>>1;
            int x=nums[mid];
            int c=query(L[i],R[i]+1,1,1,n+1,x);
            if(c>=k)
                ub=mid;
            else
                lb=mid;
        }
        printf("%d\n",nums[ub]);
    }
    scanf("%d",&n);
    return 0;
}