P4086 [USACO17DEC]My Cow Ate My Homework
思维僵化,一直觉得平均值只能二分维护。
看到这个题突然想到,直接前缀和也可以维护啊!
对于这道题来说,求出后缀和以后扫一遍,搞到最大平均值。
然后再从头扫一遍,输出与最大平均值相同的位置。
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN = 1e5 + 20;
inline int read()
{
int x = 0; char ch = getchar();
while(!isdigit(ch)) ch = getchar();
while(isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
return x;
}
int N;
int a[MAXN], sum[MAXN], mins[MAXN];
int main()
{
//freopen("p4086.in", "r", stdin);
//freopen("p4086.out", "w", stdout);
cin>>N;
for(int i = 1; i <= N; i++) a[i] = read();
for(int i = N; i >= 1; i--) sum[i] = sum[i + 1] + a[i];
mins[N] = a[N];
for(int i = N - 1; i >= 1; i--) mins[i] = min(mins[i + 1], a[i]);
double ans = 0;
for(int k = 1; k < N; k++)
ans = max(ans, (1.0 * sum[k] - mins[k]) / (1.0 * N - k));
for(int k = 2; k < N; k++)
if(fabs((1.0 * sum[k] - mins[k]) / (1.0 * N - k) - ans) <= 1e-6)
printf("%d\n", k - 1);
return 0;
}