蓝桥杯 ：区间k大数查询 (主席树)

算法训练 区间k大数查询
时间限制：1.0s 内存限制：256.0MB

问题描述
给定一个序列，每次询问序列中第l个数到第r个数中第K大的数是哪个。

输入格式
第一行包含一个数n，表示序列长度。

第二行包含n个正整数，表示给定的序列。

第三个包含一个正整数m，表示询问个数。

接下来m行，每行三个数l,r,K，表示询问序列从左往右第l个数到第r个数中，从大往小第K大的数是哪个。序列元素从1开始标号。

输出格式
总共输出m行，每行一个数，表示询问的答案。
样例输入
5
1 2 3 4 5
2
1 5 2
2 3 2
样例输出
4
2
数据规模与约定
对于30%的数据，n,m<=100；

对于100%的数据，n,m<=1000；

保证k<=(r-l+1)，序列中的数<=106。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<vector>

using namespace std;

struct Node
{
    int l;
    int r;
    int sum;
    Node()
    {
        sum = 0;
    }
}T[100007 * 40];
int cnt =0;
int n,m;
int root[100007],a[100007];
vector<int> v;
int getid(int x)
{
    return lower_bound(v.begin(),v.end(),x) - v.begin() + 1;
}
void update(int l,int r,int &x,int y,int pos)
{
    T[++cnt] = T[y];
    T[cnt].sum ++;
    x = cnt;
    if(l==r)
        return;
    int mid = (l + r)/2;
    if(pos <= mid)
        update(l,mid,T[x].l,T[y].l,pos);
    else
        update(mid + 1,r,T[x].r,T[y].r,pos);
}
int query(int l,int r,int x,int y,int k)
{
    if(l == r)
        return l;
    int mid = (l + r)/2;
    int sum = T[T[y].l].sum - T[T[x].l].sum;
    if(sum >= k)
        return query(l,mid,T[x].l,T[y].l,k);
    else
        return query(mid+1,r,T[x].r,T[y].r,k - sum);
}
int main()
{
    scanf("%d",&n);
    for(int i =1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for(int i =1;i<= n;i++)
    {
        update(1,n,root[i],root[i-1],getid(a[i]));
    }
    scanf("%d",&m);
    while(m--)
    {
        int x,y,k;
        scanf("%d%d%d",&x,&y,&k);
        if(x>y)
            swap(x,y);
        k = (y - x + 1) - k + 1;
        printf("%d\n",v[query(1,n,root[x-1],root[y],k) - 1]);
    }
    return 0;
}