题目:

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

public class Main {
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int N;
long sum=0,S=0; //sum储存数组中所有元素和,S储存最终结果。
N=sc.nextInt();
int[] arr=new int[N];
for (int i=0;i<N;i++){
arr[i]=sc.nextInt();
sum+=arr[i];
}
for (int i = 0; i < N; i++) {
S+=arr[i]*(sum-arr[i]);
sum-=arr[i];
}
System.out.println(S);
sc.close();
}
}

思路:

看到题目我第一个想到的就是暴力解法,但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%通过有时不是,看其它题解也有人反应此问题,看其它题解思路都是相同的,代码应该没问题,应该是系统问题。