首页 > 代码库 > 中庸之道(codevs 2021)

中庸之道(codevs 2021)

题目描述 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之间的整数。

技术分享
/*  可持续性线段树 求第k大 */#include<cstdio>#include<iostream>#include<algorithm>#define N 100010using namespace std;int a[N],co[N],root[N],n,m,cnt;struct node{    int lc,rc,sum;};node t[N*40];int read(){    char c=getchar();int num=0,flag=1;    while(c<0||c>9){if(c==-)flag=-1;c=getchar();}    while(c>=0&&c<=9){num=num*10+c-0;c=getchar();}    return num*flag;}int build(int v,int x,int y){    int k=++cnt;t[k].sum=v;    t[k].lc=x;t[k].rc=y;    return k;}void insert(int &root,int pre,int l,int r,int pos){    root=build(t[pre].sum+1,t[pre].lc,t[pre].rc);    if(l==r)return;    int mid=(l+r)/2;    if(pos<=mid)insert(t[root].lc,t[pre].lc,l,mid,pos);    else insert(t[root].rc,t[pre].rc,mid+1,r,pos);}int query(int x,int y,int l,int r,int k){    if(l==r)return l;    int mid=(l+r)/2;    int sum=t[t[y].lc].sum-t[t[x].lc].sum;    if(k<=sum)return query(t[x].lc,t[y].lc,l,mid,k);    else return query(t[x].rc,t[y].rc,mid+1,r,k-sum);}int main(){    freopen("jh.in","r",stdin);    n=read();m=read();    for(int i=1;i<=n;i++)    {        a[i]=read();        co[i]=a[i];    }    sort(co+1,co+1+n);    int num=unique(co+1,co+n+1)-co-1;    for(int i=1;i<=n;i++)    {        int pos=lower_bound(co+1,co+num+1,a[i])-co;        insert(root[i],root[i-1],1,num,pos);    }    for(int i=1;i<=m;i++)    {        int l=read(),r=read(),k=(l+r)/2-l+1;        int pos=query(root[l-1],root[r],1,num,k);        printf("%d\n",co[pos]);    }    return 0;}
View Code

 

中庸之道(codevs 2021)