蓝桥杯刷题 前缀和与差分-[2080]求和（C++）

题目描述

给定 n 个整数 a1, a2, · · · , an ，求它们两两相乘再相加的和，即 S = a1 · a2 + a1 · a3 + · · · + a1 · an + a2 · a3 + · · · + an-2 · an-1 + an-2 · an + an-1 · an

输入格式

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

第二行包含 n 个整数 a1, a2, · · · an。

输出格式

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

样例输入

4
1 3 6 9

样例输出

117

知识点：前缀和与差分

代码 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=200020;
ll a[N];
ll b[N];
ll sum;
int main()
{
	ll n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		b[i]=b[i-1]+a[i];
	}
	for(int i=1;i<=n-1;i++)
	{
		sum+=a[i]*(b[n]-b[i]);
	}
	cout<<sum<<endl;
	return 0;
}