静态区间第K大 (主席树) Kth number HDU - 2665

给出一个序列,询问n次,问L,R区间第k大的数

 

/*
静态区间第k小/大
主席树,也叫可持久化线段树
学习的前提条件是权值线段树,也就是计算区间内有几个数的tree,
可持久化也就是复制上一棵树的结点,然后进行修改,

1.update:
判断更新的点在左边还是右边,然后++就好了,复制结点
left == right 返回
2.query:
查询第几个,需要的是一个(1,a)区间的线段树和一个(1,b)区间的线段树
找第k小,为什么会是两个left的sum相减?
首先注意,每个线段树的区间都是相同的,只是sum的数值不同,
num = tree[b.left].sum-tree[a.left].sum 代表了左区间应该有多少数,也就是第k小是在左区间还是右区间
如果是在右区间,就用num-k,不然直接搜k
*/
#include
#include
#include
#include
#include
using namespace std;
const int maxn = 1e5 + 10;
int n,m,root[maxn],cnt;
struct node
{
    int l,r,sum;
    node()
    {
        l = r = sum = 0;
    }
} Tree[maxn*20];
vector vec;
int s[maxn];
int getID(int a)
{
    return lower_bound(vec.begin(),vec.end(),a)-vec.begin()+1;
}
void update(int Left,int Right,int &rt,int last,int ID)
{
    Tree[++cnt] = Tree[last];
    Tree[cnt].sum++;
    rt = cnt;
    if(Left == Right)return ;
    int mid = (Left+Right)>>1;
    if(ID<=mid)
        update(Left,mid,Tree[rt].l,Tree[last].l,ID);
    else update(mid+1,Right,Tree[rt].r,Tree[last].r,ID);

}
int query(int Left,int Right,int x,int y,int k)
{
    if(Left == Right)
        return Left;
    int mid = (Left+Right)>>1;
    int num = Tree[Tree[y].l].sum - Tree[Tree[x].l].sum;
    if(k<=num)
        return query(Left,mid,Tree[x].l,Tree[y].l,k);
    else return query(mid+1,Right,Tree[x].r,Tree[y].r,k-num);
}
void init()
{
    cnt = 0;
    vec.clear();
    Tree[0].l = Tree[0].r = Tree[0].sum = 0;
}
int main()
{
    ios::sync_with_stdio(false);
    //freopen("in.txt","r",stdin);
    int t;
    cin>>t;
    while(t--)
    {
        init();
        cin>>n>>m;
        for(int i=1; i<=n; i++)
        {
            cin>>s[i];
            vec.push_back(s[i]);
        }
        sort(vec.begin(),vec.end());
        vec.erase(unique(vec.begin(),vec.end()),vec.end());//离散化好评
        for(int i=1; i<=n; i++)
        {
            update(1,n,root[i],root[i-1],getID(s[i]));
        }
        for(int i=0; i>a>>b>>k;
            cout<