P8772 [蓝桥杯 2022 省 A] 求和

最新推荐文章于 2024-10-16 00:00:43 发布

zhaozhanghui

最新推荐文章于 2024-10-16 00:00:43 发布

阅读量379

点赞数 6

文章标签：蓝桥杯算法职场和发展

本文链接：https://blog.csdn.net/zhaozhanghui/article/details/135894881

版权

乍一看此题很简单，直接想都没想二重循环秒了。但果不其然超时了，只得了30分。

对于 30% 的数据, 1≤�≤1000,1≤��≤1001≤n≤1000,1≤ai≤100 。

对于所有评测用例, 1≤�≤2×105,1≤��≤10001≤n≤2×105,1≤ai≤1000 。

蓝桥杯 2022 省赛 A 组 C 题。

所以只能降低时间复杂度。可以使用前缀和求解。 $s=a1*a2+a1*a3+\cdot \cdot \cdot +a1*an+a2*a3+\cdot \cdot \cdot +a(n-2)*a(n-1)+a(n-2)*an+a(n-1)*an=a1(a2+\cdot \cdot \cdot +an)+a2(a3+\cdot \cdot \cdot +an)+\cdot \cdot \cdot +a(n-1)*an$

可以用一个sum数组储存前n项和。直接 ans+=a[i]*(sum[n-1]-sum[i]);就行。

后来又遇到了RE，数组开小了。

#include<bits/stdc++.h>
using namespace std;
#define MAXN 200000
int main()
{
    long n;
    cin>>n;
    long long a[MAXN];
    long long sum[MAXN];
    long long ans=0;
    for(int i=0;i<n;i++){
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];
    }
    for(int i=0;i<n-1;i++){
        ans+=a[i]*(sum[n-1]-sum[i]);
    }
    cout<<ans;
    return 0;
}