求和(JAVA)
题目:
给定 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
评测用例规模与约定
对于 30 %30% 的数据,1≤n≤1000,1≤ai≤100 。
对于所有评测用例, 1≤n≤200000,1≤ai≤1000 。
运行限制
最大运行时间:1s
最大运行内存: 512M
代码:
import java.util.Scanner; |
思路:
看到题目我第一个想到的就是暴力解法,但1<=n<=200000,暴力解法肯定超时,所以我们要做一下优化。
题目给的试子还可以把a1到an-1提出来变成
a1*(a2+a3+……+an)+a2*(a3+a4+……+an)+…+an-1*an
我们发现提出来后括号中的试子都可以通过sum(所有数的和)减去之前的数得到。所以我们可以通过一个for循环解题。
for (int i = 0; i < N; i++) {
S+=arr[i]*(sum-arr[i]);
sum-=arr[i];
}
注:此代码提交后有时是100%通过有时不是,看其它题解也有人反应此问题,看其它题解思路都是相同的,代码应该没问题,应该是系统问题。
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 大数据科技协会-zky!