HDU 2665 Kth number （主席树）

太过爱你忘了你带给我的痛 2022-06-16 14:28 185阅读 0赞

题目链接：  
[点击打开题目链接][Link 1]

题意：求第K大的数。

题解：裸的主席树。

AC代码：

//#include <bits/stdc++.h>
    #include <iostream>
    #include <cstdio>
    #include <algorithm>
    using namespace std;
    const int N=100010;
    int n,m;
    int a[N],b[N];
    int lson[N*20],rson[N*20],sum[N*20],root[N];
    int tot;
    void build(int l,int r,int &rt)//建立空树，节点均为 0 
    {
        rt=(++tot);
        sum[rt] = 0;
        if(l==r)return ;
        int mid =(l+r)>>1;
        build(l,mid,lson[rt]);
        build(mid+1,r,rson[rt]);
    }
    void update(int pre ,int &rt,int l,int r,int val)//sum[i]储存的是它所管辖区间内元素出现的次数，即[l,r]区间 
    {
        rt=(++tot);
        lson[rt]=lson[pre];
        rson[rt] = rson[pre];
        sum[rt] = sum[pre] + 1;
        if(l == r) return;  
        int mid = (l + r) >> 1;  
        if(val <= mid) 
        update(lson[pre], lson[rt], l, mid, val); //此元素在左区间  
        else 
        update(rson[pre], rson[rt], mid + 1, r, val);//右区间 
    }
    
    int query(int ss, int tt, int l, int r, int k) //返回值是第k大数在离散去重后的数组中的下标，线段树管理的区间是离散后的区间  
    { 
        if(l == r) return l; //此时无论k值，我们确定k就在[l,r]区间中，又l==r，结果求出  
        int mid = (l + r) >> 1;  
        int tmp = sum[lson[tt]] - sum[lson[ss]]; //左区间内元素个数
        //要查询第k大数，先求左区间内元素的个数tmp，tmp >= k，直接查询左区间内的第k大，否则查询右区间内的第k-tmp大   
        if(tmp >= k) return query(lson[ss], lson[tt], l, mid, k);
        else return query(rson[ss], rson[tt], mid + 1, r, k - tmp);  
    }  
    int main()
    {
        int t ;
        scanf("%d",&t);
        while(t--)
        {
            scanf("%d%d",&n,&m);
            tot = 0;
            for(int i=1;i<=n;i++)
            {
                scanf("%d",&a[i]);
                b[i]=a[i];
            }
            sort(b+1,b+1+n);
            int len =unique(b+1,b+1+n)-b-1;
            build(1,len,root[0]);
            for(int i=1;i<=n;i++)
            {
                a[i]=lower_bound(b+1,b+1+len,a[i])-b;
                update(root[i-1],root[i],1,len,a[i]);
            }
            while(m--)
            {
                int l,r,k;
                scanf("%d%d%d",&l,&r,&k);
                printf("%d\n",b[query(root[l-1],root[r],1,len,k)]);
            }
        }
        return 0;
    }

[Link 1]: http://acm.hdu.edu.cn/showproblem.php?pid=2665