蓝桥杯2022真题：求和

1、求和

题目描述：给定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，表示所求的和。请使用合适的数据类型进行运算。

import os
import sys
n = int(input())    #得到输入的n，转换为整型
num = list(map(int,input().split()))   #将输入的字符转化成整型数据

A = 0     #前n项和
S = 0       #最终答案
"""
假设有4个数:a1,a2,a3,a4
a1*a2+a1*a3+a1*a4+a2*a3+a2*a4+a3*a4 = (a1)*a2 + (a1+a2)*a3 + (a1+a2+a3)*a4
    ->用一个for i in range(n-1),依次求出来前1项和乘a2，前2项和乘a3，..，前n-1项和乘an,然后再把它们相加，就可以得到结果
    ->此时的时间复杂度是 n
"""
for i in range(n-1):   #利用前缀和思想，思想如上
  A += num[i]
  S += num[i+1]*A
print(S)