P1404 平均数

P1404 平均数
二分答案,问题在于怎么O(n)check,把每个数都-mid,然后求前缀和,有点像最大子段和,精度问题的话就都乘1000

#include <iostream>
#include <cstdio>
#include <queue>
#include <algorithm>
#include <cmath>
#include <cstring>
#define inf 2147483647
#define N 1000010
#define p(a) putchar(a)
#define For(i,a,b) for(long long i=a;i<=b;++i)
//by war
//2019.8.26
using namespace std;
long long n,m;
long long a[N],sum[N],Min,l,r,mid;
void in(long long &x){
    long long y=1;char c=getchar();x=0;
    while(c<'0'||c>'9'){if(c=='-')y=-1;c=getchar();}
    while(c<='9'&&c>='0'){ x=(x<<1)+(x<<3)+c-'0';c=getchar();}
    x*=y;
}
void o(long long x){
    if(x<0){p('-');x=-x;}
    if(x>9)o(x/10);
    p(x%10+'0');
}

bool check(long long x){
    Min=0;
    For(i,1,m-1)
        sum[i]=sum[i-1]+a[i]-x;
    For(i,m,n){
        sum[i]=sum[i-1]+a[i]-x;
        if(sum[i]-Min>=0)
            return 1;
        Min=min(Min,sum[i-m+1]);
    }
    return 0;
}

signed main(){
    in(n);in(m);
    For(i,1,n){
        in(a[i]);
        a[i]*=1000;
    }
    l=1;r=2000010;
    while(l<r){
        mid=(l+r)>>1;
        if(check(mid))
            l=mid+1;
        else
            r=mid;
    }
    o(l-1);
    return 0;
}

 

posted @ 2019-08-26 16:03  WeiAR  阅读(...)  评论(...编辑  收藏