LOJ 2432. 「POI2014」代理商 Couriers （静态主席树）

#2432. 「POI2014」代理商 Couriers

样例输入

7 5
1 1 3 2 3 4 3
1 3
1 4
3 7
1 7
6 6

样例输出

1
0
3
0
4

数据范围与提示

对于 3030 的数据，保证 n,m≤ 5000

对于 6565 的数据，保证 n,m≤5×10^4

对于所有数据，保证 n,m≤5×10^5.
 

思路：令k=（r-l+1）/2，如果左子区间个数大于k则在左子区间找，如果有子区间个数大于k则在右子区间找，否则返回 0 

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=500010;
int a[N],tot,n,m;
int rt[N],ls[N<<5],rs[N<<5],sum[N<<5];
void build(int &t,int l,int r)
{
    t=++tot;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(ls[t],l,mid);
    build(rs[t],mid+1,r);
}
int update(int t,int x,int l,int r)
{
    int nt=++tot;
    ls[nt]=ls[t];
    rs[nt]=rs[t];
    sum[nt]=sum[t]+1;
    if(l==r) return nt;
    int mid=(l+r)>>1;
    if(x<=mid) ls[nt]=update(ls[nt],x,l,mid);
    else rs[nt]=update(rs[nt],x,mid+1,r);
    return nt;
}
int query(int u,int v,int l,int r,int k)
{
    int mid=(l+r)>>1,ans;
    int x=sum[ls[v]]-sum[ls[u]];
    int y=sum[rs[v]]-sum[rs[u]];
    if(l==r) return l;
    if(x>k) ans=query(ls[u],ls[v],l,mid,k);
    else if(y>k) ans=query(rs[u],rs[v],mid+1,r,k);
    else ans=0;
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    tot=0;
    build(rt[0],1,n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        rt[i]=update(rt[i-1],a[i],1,n);
    }
    while(m--)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        int k=(r-l+1)>>1;
        int ans=query(rt[l-1],rt[r],1,n,k);
        printf("%d\n",ans);
    }
    return 0;
}