2022年第十三届蓝桥杯省赛真题-求和

题目描述
给定 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% 的数据，1 ≤ n ≤ 1000，1 ≤ ai ≤ 100。

对于所有评测用例，1 ≤ n ≤ 200000，1 ≤ ai ≤ 1000。

暴力破解会超时。所有样例用因式分解的平方差公式。

输出整数=((a1+a2+a3+…+an)²-a1²-a2²-a3²-…-an²)/2

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n =scanner.nextInt();
        int [] arr= new int [n];
        long [] arr1 = new long[n];
        long sum = 0;
        for (int i = 0; i <n ; i++) {
            arr[i] = scanner.nextInt();
            arr1[i]=(long)Math.pow(arr[i],2);
            sum+=arr[i];
        }
        sum =(long)Math.pow(sum,2);
        for (int i = 0; i <n ; i++) {
            sum-=arr1[i];
        }
        System.out.println(sum/2);
    }
}