P8772 [蓝桥杯 2022 省 A] 求和（前缀和）

局部地区有雨_thorough

已于 2023-12-07 21:51:56 修改

阅读量210

点赞数 7

文章标签：蓝桥杯算法

于 2023-12-07 21:49:24 首次发布

本文链接：https://blog.csdn.net/qq_63446418/article/details/134865176

版权

P8772 [蓝桥杯 2022 省 A] 求和

题目描述

给定 $n$ 个整数 $a_{1}, a_{2}, \cdots, a_{n}$ , 求它们两两相乘再相加的和，即

$S=a_{1} \cdot a_{2}+a_{1} \cdot a_{3}+\cdots+a_{1} \cdot a_{n}+a_{2} \cdot a_{3}+\cdots+a_{n-2} \cdot a_{n-1}+a_{n-2} \cdot a_{n}+a_{n-1} \cdot a_{n}$

输入格式

输入的第一行包含一个整数 $n$ 。

第二行包含 $n$ 个整数 $a_{1}, a_{2}, \cdots a_{n}$ 。

输出格式

输出一个整数 $S$ ，表示所求的和。请使用合适的数据类型进行运算。

样例 #1

样例输入 #1

4
1 3 6 9

样例输出 #1

提示

对于 $\%$ 的数据, $\leq n \leq 1000,1 \leq a_{i} \leq 100$ 。

对于所有评测用例, $\leq n \leq 2\times10^5,1 \leq a_{i} \leq 1000$ 。

蓝桥杯 2022 省赛 A 组 C 题。

解题思路1：

根据题目问题的描述，可以对数组a[ ]的每一项进行一一枚举，执行两次循环直接暴力求出sum。

问题在于：

样例的范围最大为 $\times10^5$ ，暴力法时间复杂度为O( $n^{2}$ )，并不能AC，只能过部分样例

思路1C++代码如下：（暴力枚举 30’）

#include<iostream>
using namespace std;
const int N=2e5+10;
int a[N];
long long sum=0;
int n;

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }

    for(int i=1;i<=n-1;i++)
    {
        for(int j=i+1;j<=n;j++) {
            sum+=a[i]*a[j];
        }
    }
    cout<<sum;
}

解题思路2：利用前缀和

可以把S变形一下：
$S=a_{1}\cdot\sum\limits_{i=2}^{n} a_{i} +a_{2}\cdot\sum\limits_{i=3}^{n} a_{i} +\ldots+ a_{n-2}\cdot\sum\limits_{i=n-1}^{n} a_{i} +a_{n-1}\cdot a_{n}$
可以轻松的看出各部分相似部分为： $\sum\limits_{i=k}^{n} a_{i}$
于是便联想到前缀和这一算法思想，进一步的得出前缀和： $S_k=\sum\limits_{i=1}^{k} a_{i}$ ，可知晓： $\sum\limits_{i=k}^{q} a_{i}=S_q-S_{k-1}$

所以可以最终推导出：

$S=\sum\limits_{i=1}^{n-1}( a_{i}\cdot(S_n-S_i))$

AC代码奉上：

#include<iostream>
using namespace std;
const int N=2e5+10;
int a[N];
long long sum[N];
long long s=0;
int n;

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        sum[i]=a[i]+sum[i-1];
    }

    for(int i=1;i<=n-1;i++)
    {
        s+=a[i]*(sum[n]-sum[i]);
    }
    cout<<s;
}