题目描述
给定 nn 个整数 a1,a2,⋯ ,ana1,a2,⋯,an, 求它们两两相乘再相加的和,即
S=a1⋅a2+a1⋅a3+⋯+a1⋅an+a2⋅a3+⋯+an−2⋅an−1+an−2⋅an+an−1⋅anS=a1⋅a2+a1⋅a3+⋯+a1⋅an+a2⋅a3+⋯+an−2⋅an−1+an−2⋅an+an−1⋅an
输入格式
输入的第一行包含一个整数 nn 。
第二行包含 nn 个整数 a1,a2,⋯ana1,a2,⋯an 。
输出格式
输出一个整数 S,表示所求的和。请使用合适的数据类型进行运算。
输入输出样例
输入 #1
4 1 3 6 9
输出 #1
117
说明/提示
对于 30%30% 的数据, 1≤n≤1000,1≤ai≤1001≤n≤1000,1≤ai≤100 。
对于所有评测用例, 1≤n≤2×105,1≤ai≤10001≤n≤2×105,1≤ai≤1000 。
蓝桥杯 2022 省赛 A 组 C 题。
分析
正常情况下如下:(使用前缀和,暴力容易超时)
S=a1⋅a2+a1⋅a3+⋯+a1⋅an+a2⋅a3+⋯+an−2⋅an−1+an−2⋅an+an−1⋅an=(a2+a3+⋯+an)⋅a1+(a3+a4+⋯+an)⋅a2+⋯+an⋅an−1
显然,对于任意的 ∑i=jnaii=j∑nai,其中 2≤j≤n2≤j≤n,我们都可以通过维护前缀和快速地求出来。令 sumi=sumi−1+aisumi=sumi−1+ai,则有 ∑i=jnai=sumn−sumj−1i=j∑nai=sumn−sumj−1。
#include<bits/stdc++.h>
#define MAXN 200005
int n, a[MAXN];
long long sum[MAXN], ans;
int main()
{
std::cin >> n;
for(int i = 1; i <= n; i++)
{
std::cin >> a[i];
sum[i] = sum[i - 1] + a[i];
}
for(int i = 1; i <= n - 1; i++)
ans += a[i] *(sum[n] - sum[i]);
std::cout << ans << std::endl;
return 0;
}
但你可能会说 ,嘤嘤嘤本蒟蒻看不懂一点
悄悄告诉你吧,本题也可以用完全平方公式!!!
咳咳,先从一个简单的例子入手:
在 11,22,33,44,55,66,77,88,99,1010 这些正整数中每两个数相乘的乘积之和是多少?
我们都知道这十个数两两相乘的乘积有 C102=45C102=45 个,有多项式:
(1+2+3+⋯+9+10)2−(12+22+32+⋯+92+102)22(1+2+3+⋯+9+10)2−(12+22+32+⋯+92+102)
为这十个数两两相乘的乘积之和。
- 为什么?
先看一下下面的算式:
(1+2+3+⋯+9+10)2(1+2+3+⋯+9+10)2
由完全平方公式展开后为:
(12+22+32+⋯+92+102)+2×(1×2+1×3+⋯9×10)(12+22+32+⋯+92+102)+2×(1×2+1×3+⋯9×10)
不难发现多项式 (1×2+1×3+⋯9×10)(1×2+1×3+⋯9×10) 正是要求的答案!
易得:
(1×2+1×3+⋯9×10)=(1+2+3+⋯+9+10)2−(12+22+32+⋯+92+102)2(1×2+1×3+⋯9×10)=2(1+2+3+⋯+9+10)2−(12+22+32+⋯+92+102)
那么本题的公式就是:
(a1+a2+a3+⋯+an−1+an)2−(a12+a22+a32+⋯+an2)22(a1+a2+a3+⋯+an−1+an)2−(a12+a22+a32+⋯+an2)
最后送上代码!
#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,x,mul=0,sum=0;
//mul:和的平方
//sum:平方的和
signed main(){
cin>>n;
for(int i=1;i<=n;i++){
cin>>x;
sum+=(x*x);
mul+=x;
}
cout<<(mul*mul-sum)/2;
return 0;
}