BZOJ3524Couriers

3524: [Poi2014]Couriers
Time Limit: 20 Sec Memory Limit: 128 MB
Submit: 1297 Solved: 448
Description
给一个长度为n的序列a。1≤a[i]≤n。
m组询问,每次询问一个区间[l,r],是否存在一个数在[l,r]中出现的次数大于(r-l+1)/2。如果存在,输出这个数,否则输出0。
Input
第一行两个数n,m。
第二行n个数,a[i]。
接下来m行,每行两个数l,r,表示询问[l,r]这个区间。
Output
m行,每行对应一个答案。
Sample Input
7 5
1 1 3 2 3 4 3
1 3
1 4
3 7
1 7
6 6
Sample Output
1
0
3
0
4
HINT
【数据范围】
n,m≤500000
Source
By Dzy
主席树。。
附上本蒟蒻的代码:

#include
#include
using namespace std;
int n,m,sz,root[500001],ls[10000001],rs[10000001],sum[10000001];

int read()
{
    int w=0,c=1; char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-') c=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      w=w*10+ch-'0',ch=getchar();
    return w*c;
}

void update(int l,int r,int x,int &y,int v)
{
    int mid=(l+r)/2;
    y=++sz,sum[y]=sum[x]+1;
    if (l==r) return;
    ls[y]=ls[x],rs[y]=rs[x];
    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=n,mid,x,y,tmp=(R-L+1)/2;
    x=root[L-1],y=root[R];
    while (l!=r)
      {
        if (sum[y]-sum[x]<=tmp) return 0;
        mid=(l+r)/2;
        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 0;
      }
    return l;
}

int main()
{
    int i,l,r,x;
    n=read(),m=read();
    for (i=1;i<=n;i++)
      x=read(),update(1,n,root[i-1],root[i],x);
    for (i=1;i<=m;i++)
      l=read(),r=read(),printf("%d\n",query(l,r));
    return 0;
}

你可能感兴趣的:(主席树)