传送门:Hdu 5857 Median
题意:给你一个有序的序列,取出连续的两段l1,l2,r1,r2,求这两段合并之后的中位数。
思路:比赛的时候没有看到有序,用了主席树求第k大
#include<bits/stdc++.h>
using namespace std;
const int maxn=101000;
int root[maxn],sz;
long long t[2*maxn],a[maxn];
struct node{
int l,r,w;
}T[maxn*30];
struct q{
int l1,r1,l2,r2;
}Q[maxn];
void update(int &i,int l,int r,int num){
T[++sz]=T[i],i=sz;
T[i].w++;
if(l==r)
return ;
int mid=(l+r)>>1;
if(num<=mid)
update(T[i].l,l,mid,num);
else
update(T[i].r,mid+1,r,num);
}
int query(int x,int y,int x1,int y1,int l,int r,int k){
if(l==r)
return l;
int mid=(l+r)>>1;
if(T[T[y].l].w-T[T[x].l].w+T[T[y1].l].w-T[T[x1].l].w>=k)
return query(T[x].l,T[y].l,T[x1].l,T[y1].l,l,mid,k);
else
return query(T[x].r,T[y].r,T[x1].r,T[y1].r,mid+1,r,k-(T[T[y].l].w-T[T[x].l].w+T[T[y1].l].w-T[T[x1].l].w));
}
int main(){
int _,n,q;
scanf("%d",&_);
while(_--){
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]),t[i]=a[i];
root[0]=0,sz=0;
for(int i=1;i<=q;i++)
scanf("%d%d%d%d",&Q[i].l1,&Q[i].r1,&Q[i].l2,&Q[i].r2);
sort(t+1,t+n+1);
int m=unique(t+1,t+n+1)-t-1;
for(int i=1;i<=n;i++){
int num=lower_bound(t+1,t+m+1,a[i])-t;
root[i]=root[i-1];
update(root[i],1,m,num);
}
for(int i=1;i<=q;i++){
int Count=Q[i].r1-Q[i].l1+2+Q[i].r2-Q[i].l2;
if(Count&1)
printf("%.1f\n",1.0*t[query(root[Q[i].l1-1],root[Q[i].r1],root[Q[i].l2-1],root[Q[i].r2],1,m,Count/2+1)]);
else{
long long num1=t[query(root[Q[i].l1-1],root[Q[i].r1],root[Q[i].l2-1],root[Q[i].r2],1,m,Count/2+1)];
long long num2=t[query(root[Q[i].l1-1],root[Q[i].r1],root[Q[i].l2-1],root[Q[i].r2],1,m,Count/2)];
printf("%.1f\n",1.0*(num1+num2)/2);
}
}
}
return 0;
}