HDU6278 Just H index(可持久化线段树+二分答案)

给n个数,q次询问,每次询问给出一个区间[l,r],要你求出最大的h,使得在[l,r]这个区间内满足,有h个数的值大于等于h。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+100;
const int M=maxn*40;
int a[maxn],t[maxn],T[maxn];
int lson[M],rson[M],c[M],tot;
int n,q;
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 up (int root,int p,int v) {
    int newRoot=tot++;
    int tmp=newRoot;
    c[newRoot]=c[root]+v;
    int l=1,r=n;
    while (l<r) {
        int mid=(l+r)>>1;
        if (p<=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]+v;
    }
    return tmp;
}

int query (int left_root,int right_root,int l,int r,int L,int R) {
    if (l>=L&&r<=R) return c[left_root]-c[right_root];
    int mid=(l+r)>>1;
    int ans=0;
    if (L<=mid) ans+=query(lson[left_root],lson[right_root],l,mid,L,R);
    if (R>mid) ans+=query(rson[left_root],rson[right_root],mid+1,r,L,R);
    return ans; 
}

int main () {
    while (~scanf("%d%d",&n,&q)) {
        for (int i=1;i<=n;i++) scanf("%d",a+i);
        tot=0;
        T[n+1]=build(1,n);
        for (int i=n;i;i--) T[i]=up(T[i+1],a[i],1);
        for (int i=1;i<=q;i++) {
            int l,r;
            scanf("%d%d",&l,&r);
            int ll=1,rr=n;//二分答案
            int ans=-1;
            while (ll<=rr) {
                int mid=(ll+rr)>>1;
                if (query(T[l],T[r+1],1,n,mid,n)>=mid) {
                    ans=mid;
                    ll=mid+1;
                }
                else {
                    rr=mid-1;
                }
            } 
            printf("%d\n",ans);
        }
    }
}

 

posted @ 2020-11-22 14:28  zlc0405  阅读(86)  评论(0编辑  收藏  举报