bzoj2223: [Coci 2009]PATULJCI 主席树

来源:互联网 发布:大数据应用发展趋势 编辑:程序博客网 时间:2024/06/07 04:46

第一个是位置的个数,第二个是最大值,n个数后才是询问次数m.......

用可持久化线段树记录某个区间里的个数,一开始插入的时候就要维护一个可持久化线段树。

然后二分答案,每次将x和y向目标区域移动,当x==y时就是某个特定的数的个数了。然后判断即可。

本题同3524.

#include <iostream>#include <algorithm>#include <cstring>#include <cstdio>using namespace std;int n,m,siz,rmax;int root[500010],ls[10000010],rs[10000010],sum[10000010];inline int getint(){    char ch;    while(ch=getchar(),ch<'0'||ch>'9');    int x=ch-'0';    while(ch=getchar(),ch>='0'&&ch<='9')x=x*10+(ch-'0');    return x;}void update(int l,int r,int x,int &y,int v){    y=++siz;    sum[y]=sum[x]+1;    if(l==r)return;    ls[y]=ls[x];rs[y]=rs[x];    int mid=(l+r)>>1;    if(v<=mid)update(l,mid,ls[x],ls[y],v);    else update(mid+1,r,rs[x],rs[y],v);}int query(int L,int R){    int l=1,r=rmax,mid,x,y,tmp=(R-L+1)>>1;    x=root[L-1];    y=root[R];    while(l!=r)    {        if(sum[y]-sum[x]<=tmp) return rmax+1;        mid=(l+r)>>1;        if(sum[ls[y]]-sum[ls[x]]>tmp)        {            r=mid;            x=ls[x];            y=ls[y];        }        else if(sum[rs[y]]-sum[rs[x]]>tmp)        {            l=mid+1;            x=rs[x];            y=rs[y];        }        else return rmax+1;    }    return l;}int main(){    n=getint();    rmax=getint();    for(int i=1;i<=n;i++)    {        int x;x=getint();        update(1,rmax,root[i-1],root[i],x);    }    m=getint();    for(int i=1;i<=m;i++)    {        int l,r;l=getint();r=getint();        int ans=query(l,r);        if(ans!=rmax+1) printf("yes %d\n",ans);        else printf("no\n");    }    return 0;}

0 0
原创粉丝点击