HDU6278(2018湘潭邀請賽C題Just h-index)

HDU6278

題意

在一個區間內找一個最大的h使得大於等於h的個數也得大於等於h

思路

建立一顆主席樹,然後二分找最大的h(kuangbin模板屬實好用)

代碼

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int maxx=0,tot=0,root=0;
int a[maxn],t[maxn];
int lson[40*maxn],rson[40*maxn],c[40*maxn];
int build(int l,int r){
    int root=tot++;
    c[root]=0;
    if(l!=r){
        int mid=l+r>>1;
        lson[root]=build(l,mid);
        rson[root]=build(mid+1,r);
    }
    return root;
}
int update(int root,int pos,int val){
    int newroot=tot++,tmp=newroot;
    c[newroot]=c[root]+val;
    int l=1,r=maxx;
    while(l<r){
        int mid=l+r>>1;
        if(pos<=mid){
            lson[newroot]=tot++,rson[newroot]=rson[root];
            newroot=lson[newroot];root=lson[root];
            r=mid;
        }
        else {
            rson[newroot]=tot++,lson[newroot]=lson[root];
            newroot=rson[newroot];root=rson[root];
            l=mid+1;
        }
        c[newroot]=c[root]+val;
    }
    return tmp;
}
int query(int lroot,int rroot){
    int l=1,r=maxx;
    int v=0;
    while(l<r){
        int mid=l+r>>1;
        if(v+c[rson[lroot]]-c[rson[rroot]]>=mid+1){
            lroot=rson[lroot];
            rroot=rson[rroot];
            l=mid+1;
        }
        else {
            v+=c[rson[lroot]]-c[rson[rroot]];
            lroot=lson[lroot];
            rroot=lson[rroot];
            r=mid;
        }
    }
    return l;
}
int n,m;
int main(){
    while(~scanf("%d%d",&n,&m)){
        tot=0;
        maxx=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            maxx=max(maxx,a[i]);
        }
        t[n+1]=build(1,maxx);
        for(int i=n;i>=1;i--){
            t[i]=update(t[i+1],a[i],1);
        }
        while(m--){
            int l,r;scanf("%d%d",&l,&r);
            printf("%d\n",query(t[l],t[r+1]));
        }
    }
}