bzoj3524:[Poi2014]Couriers

传送门

主席树裸题,考虑出现一半次数以上的数一定是这个区间的中位数,查询中位数再判断一下中位数出现的次数就好了
注意要离散化
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<tr1/unordered_map>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=5e5+10;tr1::unordered_map<int,int>mp;
int n,m,d[maxn],w[maxn],id,now,nid[maxn];
int rt[maxn*4],ls[maxn*20],rs[maxn*20],sum[maxn*20];
void update(int x){sum[x]=sum[ls[x]]+sum[rs[x]];}
void change(int &k,int x,int l,int r,int a)
{
    k=++id,sum[k]=sum[x],ls[k]=ls[x],rs[k]=rs[x];
    if(l==r){sum[k]++;return ;}
    int mid=(l+r)>>1;
    if(a<=mid)change(ls[k],ls[x],l,mid,a);
    else change(rs[k],rs[x],mid+1,r,a);
    update(k);
}
int get(int k,int x,int l,int r,int a,int b)
{
    if(l==r)return sum[k]-sum[x]>b?nid[l]:0;
    int mid=(l+r)>>1,now=sum[ls[k]]-sum[ls[x]];
    if(now>=a)return get(ls[k],ls[x],l,mid,a,b);
    else return get(rs[k],rs[x],mid+1,r,a-now,b);
}
int main()
{
    read(n),read(m);for(rg int i=1;i<=n;i++)read(d[i]),w[i]=d[i];sort(w+1,w+n+1);
    for(rg int i=1;i<=n;i++)if(w[i]!=w[i-1])mp[w[i]]=++now,nid[now]=w[i];
    for(rg int i=1;i<=n;i++)rt[i]=rt[i-1],change(rt[i],rt[i],1,now,mp[d[i]]);
    for(rg int i=1,x,y;i<=m;i++)read(x),read(y),printf("%d\n",get(rt[y],rt[x-1],1,now,(y-x+1)/2+1,(y-x+1)/2));
}
posted @ 2019-03-31 21:55 蒟蒻--lichenxi 阅读(...) 评论(...) 编辑 收藏