题意
给出一个长度为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;
}