农夫约翰的农场由 $N$ 块田地组成,每块地里都有一定数量的牛,其数量不会少于 $1$ 头,也不会超过 $2000$ 头。
约翰希望用围栏将一部分连续的田地围起来,并使得围起来的区域内每块地包含的牛的数量的平均值达到最大。
围起区域内至少需要包含 $F$ 块地,其中 $F$ 会在输入中给出。
在给定条件下,计算围起区域内每块地包含的牛的数量的平均值可能的最大值是多少。
输入格式
第一行输入整数 $N$ 和 $F$,数据间用空格隔开。
接下来 $N$ 行,每行输入一个整数,第 $i+1$ 行输入的整数代表第 $i$ 片区域内包含的牛的数目。
输出格式
输出一个整数,表示平均值的最大值乘以 $1000$ 再 向下取整 之后得到的结果。
数据范围
$1 \le N \le 100000$
$1 \le F \le N$
输入样例:
10 6
6
4
2
10
3
8
5
9
4
1
输出样例:
6500
#include <cmath>
#include <iostream>
using namespace std;
int n, m, cows[100010];
double l, r, sum[100010];
bool check(double avg)
{
for (int i = 1; i <= n; i ++) sum[i] = sum[i - 1] + cows[i] - avg;
double mins = 0;
for (int i = m, j = 0; i <= n; i ++, j ++)
{
mins = min(mins, sum[j]);
if (sum[i] - mins >= 0) return 1;
}
return 0;
}
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i ++)
{
cin >> cows[i];
r = max(r, (double)cows[i]);
}
while (r - l > 1e-12)
{
double mid = (l + r) / 2;
if (check(mid)) l = mid;
else r = mid;
}
cout << (int)(r * 1000 + 1e-3);
return 0;
}