序言
这是本人第一次学术类型的文章。两两乘积求和:蓝桥杯2022年这道题问懵了很多人,大家一时间都没想到用什么好的算法去做这道题,然后本题也被官方评为中等。其实一点不难,此题一出我就立马有了思路。我发现基本没人用这个思路,就在此给大家分享一下。当然最后部分我会补充暴力和前缀和的算法,读者可以自行训练熟练度。
闲言少叙,书归正传。
看题:用户登录(蓝桥官网)/ 4644. 求和 - AcWing题库(acwing链接)
新解法:
这个题一看就想起来一个公式(a+b)^2=a^2+b^2+2ab;
(a+b+c)^2 = a^2+b^2+c^2+2ab+2ac+2bc;
于是不难想到2a1a2+2a1a3+……+2a(n)a(n-1) = (a1+a2+……+an)^2-(a1^2+a2^2+……+an^2)
因此答案直接出来 所有和加起来的平方减去单个的平方和 最后除以2即为答案
O(n)的时间复杂度
相当美妙啊,学过初中数学的都应该知道。
我的代码:
#include <iostream>
using namespace std;
using ll = long long;
ll a[200005];
int main()
{
ll n,sum = 0;
ll x,y = 0,ans;
cin>>n;
for(int i = 0;i<n;i++)
{
cin>>a[i];
sum += a[i];
}
for(int i = 0;i<n;i++)
{
y += a[i]*a[i];
}
x = sum*sum;
ans = (x-y)/2;
cout<<ans<<endl;
return 0;
}
无需注释,不证自明了。就是写这个拓宽大家的思路,2分钟都不要的题目,偏偏给自己思维定势住了,想破头去用算法实现的题目其实可能很简单。
暴力:
蓝桥杯已经不是暴力为王的天下了,读者不能太过于依赖暴力解题。(当然没法AC)
很朴素,两重循环遍历之
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 200020;
ll a[N], s[N], sum;
int n;
int main()
{
cin >> n;
for(int i = 1; i <= n; i++) cin >> a[i];
for(int i = 1; i <= n; i++)
{
for(int j = i+1; j <= n; j++) sum += a[i] * a[j];
}
cout << sum;
return 0;
}
不做解释了。
前缀和:
#include <iostream>
using namespace std;
typedef long long ll;
int a[200005];
int main()
{
int n;ll sum=0,ans=0;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>a[i];
sum+=a[i];
}
for(int i=0;i<n;i++)
{
sum-=a[i];
ans+=a[i]*sum;
}
cout<<ans;
return 0;
}
实际上就相当于a[1]*a[2]+(a[1]+a[2])*a[3]+……+(a[1]+……+a[n-1])*a[n]给反过来写,其实不算很好想。
总结
反思为主,为什么应该这样做,第一反应是否限制住了你。如果是,那怎么没有想到前缀和的解法。很细节的地方也能反映你的掌握度。
以后也将持续写算法相关方面的文章,我们期待彼此的进步。加油!