P8772 [蓝桥杯 2022 省 A] 求和

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

wrs__lele

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

阅读量679

点赞数 14

文章标签：蓝桥杯算法 c++

本文链接：https://blog.csdn.net/wrs__lele/article/details/141229880

版权

题目描述

给定 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

说明/提示

对于 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;
}