蓝桥杯-求和（A组）（C语言）

题目描述

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

方法一（前缀和）（满分）

#include<stdio.h>
int a[200002] = { 0 };
long long s[200002] = { 0 };
int main()
{
    long long sum = 0;
    int n;
    scanf("%d", &n);
    int i;
    for (i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        s[i] =s[i-1]+ a[i];
    }
    for (i = 1; i <= n; i++)
    {
        sum +=(a[i]*(s[n]-s[i]));
    }
    printf("%lld\n", sum);
    return 0;
}

方法二（一部分数据超时）

#include<stdio.h>
int a[200002] = { 0 };
int main()
{
    long long sum = 0;
    int n;
    scanf("%d", &n);
    int i,j;
    for (i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
    }
    for (i = 1; i < n; i++)
    {
        for (j = i+1; j <= n; j++)
        {
            sum += a[i]*a[j];
        }
    }
    printf("%lld\n", sum);
    return 0;
}