Codeforces Round #703 (Div. 2)D. Max Median(前缀和)

关于这题,首先想了一个关于上取整中位数的求法:
大体做法如下,二分答案mid,对于1~n的所有数判断其的大小,如果大于等于mid,则令值为1,否则为1,统计前缀和,求区间时为了保证长度大于等于k,动态维护(使得长度大于等于k部分)前缀和的最小值即可.只要长度大于等于k的区间和大于等于0即可.
但是题目是下取整,在偶数的情况下并不适用,即偶数的条件要大于等于1.奇数的条件要大于等于0,那么考虑取奇数位和偶数位的最小值,就可以了。
改动如下:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 400005
int a[MAXN];
int x[MAXN];
int cmp[MAXN][2];
int main()
{
    
    
    //freopen("./tt.txt","r",stdin);
    int n,k;cin>>n>>k;
    for(int i=1;i<=n;i++)cin>>a[i];
    int l=1;
    int r=n;
    int ans=-1;
    for(int i=0;i<=k;i++)cmp[i][0]=cmp[i][1]=1e9;
    cmp[k][0]=0;
    while(l<=r)
    {
    
    
        int mid=l+r>>1;
        int cur=0;
        int flag=0;
        for(int i=1;i<=n;i++)
        {
    
    
            x[i]=(a[i]>=mid?1:-1);
            cur+=x[i];
            cmp[i+k][i%2]=min(cur,cmp[i+k-2][i%2]);
            cmp[i+k][(i%2)^1]=cmp[i+k-1][(i%2)^1];
            if(i<k)continue;
            if(cur-cmp[i][i%2]>0||cur-cmp[i][(i%2)^1]>=0)flag=i;
        }
        if(flag)l=mid+1,ans=mid;
        else r=mid-1;
    }
    cout<<ans<<endl;
}

我是hp,这题写麻烦了,其实只要大于0就行,这样就可以不分奇偶.

猜你喜欢

转载自blog.csdn.net/weixin_43353639/article/details/113884610
今日推荐