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
", k - 1);
    return 0;
}
原文地址:https://www.cnblogs.com/wsmrxc/p/9439514.html