HDU 5857 - Median ( 查询模拟 + 思路 )

题意

给出一个长度为n的递增数组A, 给出m个查找, 查找方式 : 给出l1, r1, l2, r2, 使得数组A中的A[l1], A[l1+1], ……, A[r1-1], A[r1] 和 A[l2], A[l2+1], ……, A[r2-1], A[r2]组成新的数组, 求这个新数组的中位数

For 90% of the data, n, m <= 100
For 10% of the data, n, m <= 100000

思路

给出的两个区间只有三种可能 : 相交, 包含, 相离

若有相交区域则相交区域内的数字出现次数为原来的两倍, 而相交区域后面的数字则在新序列中后移这些距离
分别模拟这三种情况, 分别找到新序列中该位置(pos)在原序列中对应的位置即可.

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 1e5+5;
int a[maxn];
int n, l1, l2, r1, r2;

int median( int pos ){
    if( l2 <= r1 ){
        if( r2 <= r1 ){ // 包含
            if( pos <= l2-(l1-1) ) return a[pos+l1-1];
            else if( pos <= (r2-l2+1)*2+(l2-l1) ) return a[l2+(pos-(l2-l1+1))/2];
            else return a[r2+(pos-((r2-l2+1)*2+(l2-l1)))];
        }
        else{ // 相交
            if( pos <= l2-(l1-1) ) return a[pos+l1-1];
            else if( pos <= (r1-l2+1)*2+(l2-l1) ) return a[l2+(pos-(l2-l1+1))/2];
            else return a[r1+(pos-((r1-l2+1)*2+(l2-l1)))];
        }
    }
    else{ // 相离
        if( pos <= (r1-l1+1) ) return a[pos+l1-1];
        else return a[l2+(pos-(r1-l1+1))-1];
    }
}

int main()
{
    int T, m;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d", &n, &m);
        for( int i = 1; i <= n; i++ )
            scanf("%d",&a[i]);
        while(m--){
            scanf("%d%d%d%d", &l1, &r1, &l2, &r2);
            if( l1 > l2 ){
                swap(l1, l2);
                swap(r1, r2);
            }
            int len = r1-l1+1+r2-l2+1;
            if( len % 2 != 0 )
                printf("%.1f\n", 1.0*median(len/2+1));
            else{
                //cout << median(len/2) << endl;
                //cout << median(len/2+1) << endl;
                printf("%.1f\n", 0.5*median(len/2) + 0.5*median(len/2+1) );
            }
        }
    }
    return 0;
}