51nod1685 第K大区间2

链接：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1685

题意：中文题。

分析：二分答案，然后判断大于等于当前答案的数是否有>=k个奇区间中位数是大于等于当前答案的。比赛的时候没想到这么快速判中位数大于等于当前答案，再多想想其实就出来了。设当前答案为x，那我们将所有>=x的数标为1，<x的数标为0，那么我们只需要找有多少个区间[i,j]使得sum[j]-sum[i-1]>(j-i)/2，i和j不同奇偶，即2*sum[j]-j>2*sum[i]-i，设c[i]=2*sum[i]-i。我们分奇偶设两个树状数组就能在nlogn的时间统计完当前x的所有奇区间啦。那么总的时间复杂度为O(n*logn*logn)

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=100010;
const int mod=100000000;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int INF=1000000010;
const ll MAX=1000000000000;
const double pi=acos(-1.0);
typedef double db;
typedef unsigned long long ull;
int n,a[N],b[N],c[N],sum[N],f[2][3*N];
void add(int rev,int x,int y) {
    if (x<=0) return ;
    for (;x<=3*n;x+=x&-x) f[rev][x]+=y;
}
int getsum(int rev,int x) {
    if (x<=0) return 0;
    int ret=0;
    for (;x;x-=x&-x) ret+=f[rev][x];
    return ret;
}
ll stat(int x) {
    ll ret=0;
    for (int i=1;i<=n;i++) {
        sum[i]=sum[i-1]+(a[i]>=x);
        c[i]=2*sum[i]-i+n;
    }
    memset(f,0,sizeof(f));add(0,n,1);
    for (int i=1;i<=n;i++) {
        add(i&1,c[i],1);
        ret+=(ll)getsum((i+1)&1,c[i]-1);
    }
    return ret;
}
int main()
{
    int i,l,r,mid;
    ll k;
    scanf("%d%lld", &n, &k);
    for (i=1;i<=n;i++) scanf("%d", &a[i]),b[i]=a[i];
    sort(b+1,b+n+1);
    l=1;r=n+1;mid=(l+r)>>1;
    while (l+1<r)
    if (stat(b[mid])>=k) { l=mid;mid=(l+r)>>1; }
    else { r=mid;mid=(l+r)>>1; }
    printf("%d\n", b[l]);
    return 0;
}