考试题 树状数组查找以比x大的数作为中位数的个数+二分中位数

Problem II
定义一个长度为奇数的区间的值为其所包含的的元素的中位数
现给出n个数，求将所有长度为奇数的区间的值排序后，第K大的值为多少


第一行两个数n和k(1<=n<=100000,k<=奇数区间的数量)
第二行n个数，0<=每个数<2^31
请输出一个数表示答案


Sample Input
4 3
3 1 2 4

Sample Output

2

这道题其实学长在上午的时候讲过一道类似的，只是把中位数改成了众数，在哪一道题中，同样也是二分答案但是在check的时候直接是维护两个l和r指针，然后滑动指针，找到满足条件的指针，那么这两个指针中间的所有的区间均满足条件。所以下午想的时候也是按照这样的一个l，r的思想去想的，最后也就gg了。其实在以前也做过一些关于中位数判断的题目大体也是差不多的，如何验证以这个数以及比他大的数作为中位数的区间个数。首先我们县预处理出另外一个数组，在原来数组中比这个大和等于的数重新赋值为1，而小于的就赋值为-1，然后维护一个前缀和（注意可能为负数），根据前面说的我们需要求出以这个数以及比他大的数作为中位数的区间的中位数，等价于sum[r]-sum[l]>=0的个数，这个用一个树状数组就可以乱搞出来了，同时注意题目要求的必须是奇数区间，，有这样一个事实就是： 奇数-偶数=奇数    偶数-奇数=奇数     所以只需要把下标为奇数的和偶数的分别记录下来，再分别减就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 100020
#define ll long long
using namespace std;
ll n,c[maxn*5],d[maxn*5],k;
ll cur[maxn];
void init(){
	freopen("B.in","r",stdin);
	freopen("B.out","w",stdout);
}
void update(ll x,ll add,ll flag){//1是偶数 
	if(flag==1)while(x<=200000){c[x]+=add;x+=x&-x;}
	else while(x<=200000){d[x]+=add;x+=x&-x;}
}
ll query(ll x,ll flag){
	ll ans=0;
	if(flag==1)while(x>0){ans+=c[x];x-=(x&-x);}
	else while(x>0){ans+=d[x];x-=(x&-x);}
	return ans;
}
ll t[maxn];
bool check(ll x){//检验答案 
	memset(d,0,sizeof(d));
	memset(c,0,sizeof(c));
	memset(t,0,sizeof(t));
	ll ans=0;
	t[0]=maxn;//有可能下标为负数所以先加上maxn 
	for(ll i=1;i<=n;i++)t[i]=t[i-1]+(x>cur[i]?-1:1);//预处理出新的数组 
	for(ll i=1;i<=n;i++)if(cur[i]>=x)ans++;//长度是一的也算 
	for(ll i=1;i<=n;i++){
		if(i&1){
			ans+=query(t[i],0);
			update(t[i],1,1);
		}
		else {
			
			ans+=query(t[i],1);
			update(t[i],1,0);
		}
	}
	return ans>=k;
}


int main(){
	init();
	scanf("%I64d%I64d",&n,&k);
	for(ll i=1;i<=n;i++)scanf("%d",cur+i);
	ll l=0,r=1e9;
	while(l<r){//二分答案 
		ll mid=l+r+1>>1;
		if(check(mid))l=mid;
		else r=mid-1;
	}
	printf("%I64d",l);
	return 0;
}