Codevs 2021 中庸之道(主席树)

2021 中庸之道
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 钻石 Diamond
题目描述 Description
给定一个长度为N的序列，有Q次询问，每次询问区间[L,R]的中位数。
数据保证序列中任意两个数不相同，且询问的所有区间长度为奇数。
输入描述 Input Description
第一行为N，Q。
第二行N个数表示序列。
接下来Q行，每行为L，R，表示一次询问。
输出描述 Output Description
输出Q行，对应每次询问的中位数。
样例输入 Sample Input
5 3
1 4 8 16 2
1 5
3 5
3 3
样例输出 Sample Output
4
8
8
数据范围及提示 Data Size & Hint
40%的数据，N，Q≤100；
70%的数据，N≤100；
100%的数据，N≤1000，Q≤100000，序列中的元素为1到10^9之间的整数。

/*
主席树.
区间((x+y)>>1)-x+1大值.
*/
#include<iostream>
#include<cstdio>
#include<algorithm>
#define MAXN 100001
using namespace std;
struct data{int l,r,sum,lc,rc;}tree[MAXN*4];
int n,m,root[MAXN],a[MAXN],s[MAXN],tot,cut,x,y;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x*f;
}
int build(int sum,int l,int r)
{
    int k=++cut;tree[k].sum=sum,tree[k].lc=l,tree[k].rc=r;
    return k;
}
void add(int &now,int last,int l,int r,int p)
{
    now=build(tree[last].sum+1,tree[last].lc,tree[last].rc);
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(p<=mid) add(tree[now].lc,tree[last].lc,l,mid,p);
    else add(tree[now].rc,tree[last].rc,mid+1,r,p);
}
int query(int L,int R,int p,int l,int r)
{
    if(l==r) return r;
    int ans=tree[tree[R].lc].sum-tree[tree[L].lc].sum;
    int mid=(l+r)>>1;
    if(ans>=p) return query(tree[L].lc,tree[R].lc,p,l,mid);
    else return query(tree[L].rc,tree[R].rc,p-ans,mid+1,r);
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;i++) a[i]=read(),s[i]=a[i];
    sort(s+1,s+n+1);
    tot=unique(s+1,s+n+1)-s-1;
    for(int i=1;i<=n;i++)
    {
        int p=lower_bound(s+1,s+tot+1,a[i])-s;
        add(root[i],root[i-1],1,tot,p);
    }
    int k; 
    for(int i=1;i<=m;i++)
    {
        x=read(),y=read();k=((x+y)>>1)-x+1;
        int p=query(root[x-1],root[y],k,1,tot);
        printf("%d\n",s[p]);
    }
    return 0;
}