BZOJ5178: [Jsoi2011]棒棒糖

【传送门:BZOJ5178


简要题意:

  给出n个数,给出m个询问,每个询问输入l,r,求出是否存在有一个数在l到r中出现的次数超过总数/2次,如果存在则输出这个数,否则输出0


题解:

  主席树例题,直接将n个数一个个插入主席树,然后判断是否存在一棵子树的数的数量超过总数/2次,然后继续向下找就可以了


参考代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
using namespace std;
struct node
{
    int lc,rc,c;
}tr[9100000];int trlen;
int a[510000],s[510000];
int rt[510000],n;
int LS(int d)
{
    int l=1,r=n,ans;
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(s[mid]<=d)
        {
            ans=mid;
            l=mid+1;
        }
        else r=mid-1;
    }
    return ans;
}
void Link(int &u,int l,int r,int p)
{
    if(u==0) u=++trlen;
    tr[u].c++;
    if(l==r) return ;
    int mid=(l+r)/2;
    if(p<=mid) Link(tr[u].lc,l,mid,p);
    else Link(tr[u].rc,mid+1,r,p);
}
void Merge(int &u1,int u2)
{
    if(u1==0){u1=u2;return ;}
    if(u2==0) return ;
    tr[u1].c+=tr[u2].c;
    Merge(tr[u1].lc,tr[u2].lc);
    Merge(tr[u1].rc,tr[u2].rc);
}
int ans;
int findhalf(int u1,int u2,int l,int r,int k)
{
    if(l==r) return l;
    int mid=(l+r)/2;
    if(tr[tr[u1].lc].c-tr[tr[u2].lc].c>k) return findhalf(tr[u1].lc,tr[u2].lc,l,mid,k);
    else if(tr[tr[u1].rc].c-tr[tr[u2].rc].c>k) return findhalf(tr[u1].rc,tr[u2].rc,mid+1,r,k);
    else return 0;
}
int main()
{
    int m;
    scanf("%d%d",&n,&m);
    trlen=0;memset(rt,0,sizeof(rt));
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),s[i]=a[i];
    sort(s+1,s+n+1);
    for(int i=1;i<=n;i++)
    {
        Link(rt[i],1,n,LS(a[i]));
        Merge(rt[i],rt[i-1]);
    }
    for(int i=1;i<=m;i++)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        ans=0;
        printf("%d\n",s[findhalf(rt[r],rt[l-1],1,n,(r-l+1)/2)]);
    }
    return 0;
}

 

posted @ 2018-04-17 19:02  Star_Feel  阅读(253)  评论(0编辑  收藏  举报