BZOJ 3524 [Poi2014]Couriers 主席树

https://darkbzoj.cf/problem/3524

 

给一个长度为n的序列a。1≤a[i]≤n。

m组询问，每次询问一个区间[l,r]，是否存在一个数在[l,r]中出现的次数大于(r-l+1)/2。如果存在，输出这个数，否则输出0。

 

主席树查找裸题，维护出现次数

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define LL long long
const int maxn=500010;
int n,m;
int rt[maxn]={};
int lc[maxn*20]={},rc[maxn*20]={},siz[maxn*20]={},tot=0;
void build(int &x,int y,int l,int r,int z){
    x=++tot;siz[x]=siz[y]+1;
    if(l==r)return;
    lc[x]=lc[y];rc[x]=rc[y];
    int mid=(l+r)/2;
    if(z<=mid)build(lc[x],lc[y],l,mid,z);
    else build(rc[x],rc[y],mid+1,r,z);
}
int getans(int x,int y,int l,int r,int temp){
    if(l==r)return l;
    int mid=(l+r)/2;
    if(siz[lc[y]]-siz[lc[x]]>temp) return getans(lc[x],lc[y],l,mid,temp);
    else if(siz[rc[y]]-siz[rc[x]]>temp) return getans(rc[x],rc[y],mid+1,r,temp);
    else return 0;
}
int main(){
    int x,y,temp;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&x);rt[i]=rt[i-1];
        build(rt[i],rt[i],1,n,x);
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d",&x,&y);temp=(y-x+1)/2;
        printf("%d\n",siz[rt[y]]-siz[rt[x-1]]<=temp?0:getans(rt[x-1],rt[y],1,n,temp));
    }
    return 0;
}