【YBT高效进阶】1基础算法/3二分算法/3最大均值
内存限制:256 MiB
时间限制:1000 ms
标准输入输出
题目类型:传统
评测方式:文本比较
题目描述
给定正整数序列 A A A ,求一个平均数最大的,长度不小于 L L L 的(连续的)子段。
输入格式
第一行两个整数 N N N 和 L L L。
接下来 N N N 行,每行输入一个正整数 A i A_i Ai。
输出格式
输出一个整数,表示平均值的最大值乘以 1000 1000 1000 再向下取整之后得到的结果。
样例
样例输入
10 6
6
4
2
10
3
8
5
9
4
1
样例输出
6500
数据范围与提示
对于 100% 的数据 1 < = n < = 1 0 5 , 1 < = L < = N , 1 < = A i < = 2000 1<=n<=10^5,1<=L<=N,1<=A_i<=2000 1<=n<=105,1<=L<=N,1<=Ai<=2000。
思路
二分答案
实数二分
每个数-mid
求前缀和
算出长度>=L的最大子段
若和>=0 l=mid
若何<0 r=mid
代码
#include<iostream>
#include<cstdio>
using namespace std;
double a[200010],sum[200010];
int n,L;
bool check(double x)
{
double ans=-10000000000,mn=10000000000;
int i;
for(i=1;i<=n;i++)//前缀和
sum[i]=sum[i-1]+a[i]-x;
for(i=L;i<=n;i++)
{
mn=min(mn,sum[i-L]);//当前最小值
ans=max(ans,sum[i]-mn);//最大子段和
}
return ans>=0;
}
int main()
{
double l,r,mid;
int i;
ios::sync_with_stdio(false);
for(cin>>n>>L,i=1;i<=n;i++)
cin>>a[i];
for(l=-1000000,r=1000000;l+0.00001<r;)//二分答案
{
mid=(l+r)/2;
if(check(mid))l=mid;
else r=mid;
}
cout<<int(r*1000)<<endl;
return 0;
}