Preliminaries for Benelux Algorithm Programming Contest 2019(i)

Purpose: 给定数组 a,求公式的最大值。
Solve:求 a 数组平方的前缀和和求 a 数组后缀和,遍历一遍即可。

#include <iostream>
#include <cmath>
#define ll long long

using namespace std;

ll f[1000005],e[1000005];
int main()
{
    int n,a[1000005];
    cin>>n;
    for(int i = 1; i <= n; i++)
        cin>>a[i];
    for (int i = 1; i <= n; i++)//前缀和求每个位置之前的平方和。
        f[i] = f[i - 1] + a[i] * a[i];
	for (int i = n; i >= 1; i--)//后缀和,求每个位置之后的各数之和。
        e[i] = e[i + 1] + a[i];
    ll ans = 0;
    for(int i = 1; i < n; i++)//遍历找最大值
    {
        ans = max(ans , f[i] * e[i+1]);
    }
    cout<<ans<<endl;
    return 0;
}

Cu1
发布了30 篇原创文章 · 获赞 2 · 访问量 963

猜你喜欢

转载自blog.csdn.net/CUCUC1/article/details/104866927