题目描述
给定一个长度为 n n n 的正整数序列 A A A 。求一个平均数最大的,长度不小于 L L L 的子序列。
输入格式
第一行, n n n 和 L L L ;
n n n 个正整数 A i A_i Ai,表示序列 A A A 。
输出格式
一个整数,表示答案的 1000 1000 1000 倍(不用四舍五入,直接输出)。
输入样例
10 6
6 4 2 10 3 8 5 9 4 1
输出样例
6500
数据范围
n ≤ 100000 n \leq 100000 n≤100000
解题思路
二分答案(浮点数)
标准程序
#include <cstdio>
#include <iostream>
using namespace std;
const int N = 100005;
int n, L;
double a[N], s[N];
//判平均数x 是否能满足题意要求
//将序列长度不小于L的序列减去平均数后,序列和应该大于等于0
bool check(double x) {
//将序列中所有的数减去平均数的前缀和 s
for (int i = 1; i <= n; i++) {
s[i] = s[i - 1] + a[i] - x;
}
//优化 使用minv保存(1~i-L)范围的最小值
//因为每次i增加,只增加一个数进入(1~i-L)范围,所以minv只需要和s[i-L]比较即可
double ans = -1e9, minv = 1e9;
for (int i = L; i <= n; i++) {
minv = min(minv, s[i - L]); //获取[1~i-L]中最小值
ans = max(ans, s[i] - minv); //求不小于L的序列的最大值
}
return ans >= 0; //返回判断结果
}
int main() {
scanf("%d%d", &n, &L);
for (int i = 1; i <= n; i++) scanf("%lf", &a[i]);
//浮点数二分模板
double eps = 1e-5, l = -1e6, r = 1e6;
while (r - l > eps) {
double mid = (l + r) / 2;
if (check(mid))
l = mid;
else
r = mid;
}
printf("%d", int(r * 1000));
return 0;
}