[BZOJ3524][Poi2014]Couriers（主席树）

=== ===

这里放传送门

=== ===

题解

区间内符合要求的数只可能有一个，并且如果存在这个数，那么第$(r−l)/2+1$个数一定是它。所以用主席树查出区间内第$(r−l)/2+1$个数，然后再查一下它的出现次数是不是大于$(r−l)/2$即可。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,r[500010],tot,b[500010];
struct tree{
    int l,r,val;
}t[10000000];
void update(int i){
    t[i].val=t[t[i].l].val+t[t[i].r].val;
}
void insert(int rt,int &i,int l,int r,int x){
    i=++tot;t[i]=t[rt];
    t[i].val++;
    if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) insert(t[rt].l,t[i].l,l,mid,x);
    else insert(t[rt].r,t[i].r,mid+1,r,x);
    update(i);
}
int asknum(int i,int j,int l,int r,int k){
    if (l==r) return l;//查询区间内排名为k的数
    int mid=(l+r)>>1,tmp=0;
    tmp=t[t[j].l].val-t[t[i].l].val;
    if (tmp>=k) return asknum(t[i].l,t[j].l,l,mid,k);
    else return asknum(t[i].r,t[j].r,mid+1,r,k-tmp);
}
int askval(int i,int j,int l,int r,int x){
    if (l==r) return t[j].val-t[i].val;//查询这个数在区间内出现了几次
    int mid=(l+r)>>1;
    if (x<=mid) return askval(t[i].l,t[j].l,l,mid,x);
    else return askval(t[i].r,t[j].r,mid+1,r,x);
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&b[i]);
    for (int i=1;i<=n;i++)
      insert(r[i-1],r[i],1,n,b[i]);
    for (int i=1;i<=m;i++){
        int L,R,pos,k;
        scanf("%d%d",&L,&R);
        pos=(R-L+1)/2+1;
        k=asknum(r[L-1],r[R],1,n,pos);
        if (askval(r[L-1],r[R],1,n,k)>=pos) printf("%d\n",k);
        else printf("0\n");
    }
    return 0;
}