【ACM】HDU6278 Just h-index 主席树+二分

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6278

题目大意:

求给定区间内最大的h,满足区间内至少有h个数不小于h

思路:

对于每一个查询,由于h的大小必定在[0,区间长度len]范围内,所以可以二分枚举h的值,利用主席树查找到区间内第len-h+1小的数和当前h比较,若比h小,更新上界,反之则更新下界。

代码:

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
const int maxn=200005;
int n,m;
int v[maxn],b[maxn],t[maxn],tsize,tot=0;
struct charitree
{
    int l,r,tsize;
} a[maxn*30];
int build(int l,int r)
{
    int root=tot++;
    a[root].tsize=0;
    if(l==r)
        return root;
    int m=(l+r)>>1;
    a[root].l=build(l,m);
    a[root].r=build(m+1,r);
    return root;
}
int update(int root,int x)
{
    int now=tot++;
    int tmp=now;
    a[now].tsize=a[root].tsize+1;
    int l=1,r=tsize;
    while(l<r)
    {
        int m=(l+r)>>1;
        if(x<=m)
        {
            a[now].l=tot++;
            a[now].r=a[root].r;
            now=a[now].l;
            r=m;
            root=a[root].l;
        }
        else
        {
            a[now].l=a[root].l;
            a[now].r=tot++;
            now=a[now].r;
            l=m+1;
            root=a[root].r;
        }
        a[now].tsize=a[root].tsize+1;
    }
    return tmp;
}
int ask(int lx,int rx,int k)
{
    int l=1,r=tsize;
    while(l<r)
    {
        int m=(l+r)>>1;
        if(a[a[rx].l].tsize-a[a[lx].l].tsize>=k)
        {
            r=m;
            lx=a[lx].l;
            rx=a[rx].l;
        }
        else
        {
            l=m+1;
            k-=a[a[rx].l].tsize-a[a[lx].l].tsize;
            lx=a[lx].r;
            rx=a[rx].r;
        }
    }
    return l;
}
void hash1()
{
    sort(b+1,b+1+n);
    tsize=unique(b+1,b+1+n)-b-1;//去重函数
}
int find_index(int x)
{
    return lower_bound(b+1,b+1+tsize,x)-b;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int i,j,l,r,k;
    while(~scanf("%d%d",&n,&m))
    {
        for(i=1; i<=n; i++)
            scanf("%d",&v[i]),b[i]=v[i];
        hash1();
        t[0]=build(1,tsize);
        for(i=1; i<=n; i++)
            t[i]=update(t[i-1],find_index(v[i]));
        while(m--)
        {
            scanf("%d%d",&l,&r);
            int len=r-l+1;
            int high=len,low=0,mid=0;
            while(low<=high)
            {
                mid=(low+high)/2;
                int midres=b[ask(t[l-1],t[r],len-mid+1)];
                //printf("*%d %d %d\n",low,high,midres);
                if(midres>=mid)
                    low=mid+1;
                else
                    high=mid-1;
            }
            printf("%d\n",high);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41279172/article/details/97521437