bzoj 3524: [Poi2014]Couriers（主席树）

3524: [Poi2014]Couriers

Time Limit: 20 Sec   Memory Limit: 128 MB
Submit: 1448   Solved: 507
[ Submit][ Status][ Discuss]

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

[ Submit][ Status][ Discuss]

题解：先建立静态主席树，然后区间查询[l,r]中第(r-l+1)/2+1小的数，在查询该点的出现次数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 500003
using namespace std;
int n,m;
int a[N],sz,ls[N*20],rs[N*20],sum[N*20];
int root[N];
void insert(int &i,int k,int l,int r,int x)
{
	i=++sz; ls[i]=ls[k]; rs[i]=rs[k];
	int mid=(l+r)/2;
	sum[i]=sum[k]+1;
	if (l==r) return;
	if (x<=mid)
	 insert(ls[i],ls[k],l,mid,x);
	else
	 insert(rs[i],rs[k],mid+1,r,x);
}
int query(int x,int y,int l,int r,int k)
{
	int t=sum[ls[y]]-sum[ls[x]];
	int mid=(l+r)/2;
	if (l==r)  return l;
	if (t>=k)
	 return query(ls[x],ls[y],l,mid,k);
	else
	 return query(rs[x],rs[y],mid+1,r,k-t);
}
int point(int x,int y,int l,int r,int v)
{
	int mid=(l+r)/2;
	if (l==r) return sum[y]-sum[x];
	if (v<=mid)
	 return point(ls[x],ls[y],l,mid,v);
	else
	 return point(rs[x],rs[y],mid+1,r,v);
}
int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	 scanf("%d",&a[i]);
    for (int i=1;i<=n;i++)
      insert(root[i],root[i-1],1,n,a[i]);
    for (int i=1;i<=m;i++)
     {
     	int l,r;
        scanf("%d%d",&l,&r);
        int t=(r-l+1)/2;  
        int x=query(root[l-1],root[r],1,n,t+1);  
        int ans=point(root[l-1],root[r],1,n,x); 
        if (ans>t)
         printf("%d\n",x);
        else
         printf("0\n");
     }
}