蓝桥杯——求和（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;

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%通过有时不是，看其它题解也有人反应此问题，看其它题解思路都是相同的，代码应该没问题，应该是系统问题。