牛客网D1T1中位数(二分)

题目链接

题目：

题目描述：

小N得到了一个非常神奇的序列A。这个序列长度为N，下标从1开始。A的一个子区间对应一个序列，可以由数对[l,r]表示，代表A[l], A[l + 1], …, A[r]这段数。对于一个序列B[1], B[2], …, B[k]，定义B的中位数如下：
1. 先对B排序。得到新的序列C。
2. 假如k是奇数，那么中位数为。假如k为偶数，中位数为。
对于A的所有的子区间，小N可以知道它们对应的中位数。现在小N想知道，所有长度>=Len的子区间中，中位数最大可以是多少

样例输入:

11 3

4864 8684 9511 8557 1122 1234 953 9819 101 1137 1759

样例输出：

8684

数据范围：

30%: n <= 200

60%: n <= 2000

另外有20%：不超过50个不同的数

100%：1<=Len<=n<=10^5, 1 <= a[i] <= 10^9

题解:

我由中位数联想到一个做法：枚举答案，把比答案小的数标记成-1，比答案大的数标记成1，选一个包含答案的区间，加点前缀和就可以判断合不合法。但是仔细一想，第一等于的情况不好处理，第二区间也不满足单调性，第三想二分答案却也不单调。我本来就在比赛结束前半小时才开始，这下直接放弃。

其实如果我们思路灵活一点，可以这样想，既然我们的目的是要答案最大，那么我们只要确定可以得到的答案大于等于枚举的就可以，这样就同时解决了三个问题。

正解：二分答案，比答案小的数标记成-1，其余标记成1，求出前缀最大值。如果最大区间的值大于0，就证明可以得到比当前枚举的答案大的答案。注意是大于0，当区间长度为2k+1时，显然必须大于。当区间长度为2k时，中位数为第k个数（排序后），如果等于0，那么1-k都小于枚举的答案，不成立。

代码:

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
const int N=1e5+5;
int n,len,l,r;
int ai[N],qian[N],Min[N],bi[N];

bool Judge(int x)
{
    for(int i=1;i<=n;i++)
        qian[i]=qian[i-1]+(ai[i]>=x?1:-1);
    Min[1]=qian[1];
    for(int i=2;i<=n;i++)
        Min[i]=min(Min[i-1],qian[i]);
    int J=qian[len];
    for(int i=len+1;i<=n;i++)
        J=max(J,qian[i]-Min[i-len]);
    return J>0;
}

int main()
{
    scanf("%d%d",&n,&len);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&ai[i]);
        bi[i]=ai[i];
    }
    sort(bi+1,bi+1+n);
    l=1,r=n;
    while(l<r)
    {
        if(l+1==r){
            l=(Judge(bi[r])?r:l);
            break;
        }
        int mid=((l+r)>>1);
        if(Judge(bi[mid]))
            l=mid;
        else
            r=mid-1;
    }
    printf("%d\n",bi[l]);
    return 0;
}