Hdu 5857 Median(给你一个有序的序列,取出连续的两段l1,l2,r1,r2,求这两段合并之后的中位数)

传送门: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;
}