解题思路
最直白的思路应该是暴力搜索,算出每一个长度大于L的子列的平均值,最后输出最大平均值,但是这样肯定会超时。下面附上79分暴力算法:
#include <stdio.h>
#include <algorithm>
using namespace std;
const int MAXN = 1e5 + 5;
int n, l;
int ans = 0;
int a[MAXN], sum[MAXN];
int main()
{
scanf("%d %d", &n, &l);
for (int i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
sum[i] = sum[i - 1] + a[i]; //求前缀和
}
for (int len = l; len <= n; len++)
{
for (int l = 1; l + len - 1 <= n; l++)
{
int r = l + len - 1;
ans = max(ans, (sum[r] - sum[l - 1]) * 1000 / len);
}
}
printf("%d\n", ans);
return 0;
}
观察数据规模,由于所有数据大小均不超过2000,因此本题适合用二分法进行求解。关键在于,如何判断一个子列平均值大于给定均值?
首先进行预处理,让数列里每一个数减去该平均值,只要转化后的数列任意子列和为正就满足条件,不再需要比较平均数
初始化former变量为0,利用former变量记录数列前(i-l)项和的最小值。(满足最小子串)
从i = l开始遍历,依次检查前i项和,减去former是否大于0。
原理大概是这样:想像一个函数,横坐标代表项数i,纵坐标代表前i项和,那么这个数列任意子列的和都可以通过这个函数上两点纵坐标作差得到,那么问题就转化为求这个函数的值域。
函数的值域只跟最大值与最小值有关,而之前所说的former变量记录的就是当i<=l时函数的最小值,此时只需要求前i项的最大前缀和,做差即可求出平均数最大子列。
int A[100005], a[100005];
long long b[100005];
int n, l;
bool check(const int& mid)
{
b[0] = 0;
for(int i = 1; i <= n; i++)
{
a[i] = A[i]-mid; //预处理,转化问题,只要任意子列和为正即可,不再需要求平均数
b[i] = b[i-1]+a[i]; //求前缀和
}
long long former = 0; //用long long类型防止溢出
for(int i = l; i <= n; i++) //保证子列长度大于l
{
former = min(former, b[i-l]); //前i-l项的最小子列前缀和
long long temp = b[i]-former; //前i项最大子列前缀和
if(temp >= 0)
return true;
}
return false;
}
完整代码
#include <iostream>
#include <algorithm>
using namespace std;
int A[100005];
int a[100005];
long long b[100005];
int n, l;
bool check(const int& mid)
{
b[0] = 0;
for(int i = 1; i <= n; i++)
{
a[i] = A[i]-mid;
b[i] = b[i-1]+a[i];
}
long long former = 0; //这一句要放在循环外面!
for(int i = l; i <= n; i++)
{
former = min(former, b[i-l]);
long long temp = b[i]-former;
if(temp >= 0)
return true;
}
return false;
}
int main()
{
cin >> n >> l;
for(int i = 1; i <= n; i++)
{
int temp;
cin >> temp;
A[i] = temp*1000;
}
int left = 0;
int right = 2000000;
for(; left <= right;)
{
int mid;
mid = (left+right)>>1;
if(check(mid))
left = mid+1;
else
right = mid-1;
}
cout << right << endl;
return 0;
}