洛谷——P1404 平均数

P1404 平均数

题目描述

给一个长度为n的数列，我们需要找出该数列的一个子串，使得子串平均数最大化，并且子串长度>=m。

前缀和+二分答案

#include<iostream>
#include<cstdio>
#include<cstdlib>

#define N 1000006
#define inf 0x7fffffff
#define eps 1e-5
using namespace std;

int n,m;
double a[N],b[N],sum[N];

bool check(double X){
    for(int i=;i<=n;i++)
        b[i]=a[i]-X,sum[i]=sum[i-]+b[i];
    double minn=;
    for(int i=m;i<=n;i++){
        minn=min(minn,sum[i-m]);
        if(double(sum[i]-minn)+eps>)
            return true;
    }
    return false;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++) scanf("%lf",&a[i]);

    double l=,r=2000.00,mid;
    for(int i=;i<=;i++){
        mid=(l+r)/2.0;
        if(check(mid)) l=mid;
        else r=mid;
    }

    printf("%d",(int)(l*));

    return ;
}