P8772 [蓝桥杯 2022 省 A] 求和

时间:2024-12-18 14:56:45

题目描述:

解题思路:

首先这题我们可以直接用两个for循环嵌套来控制两个变量来求值,但是这样做时间复杂度高。这里我们用到了一个前缀和差的方法。通过for循环变量第一个变量,用和差的方法的到第二个量,这样就只用了一次循环,大大减少了时间复杂度。需要注意的是,这题有时间限制开辟大数组时需要开在全局中(程序留给全局的空间大,而留给main函数内的空间少)。

样例代码:

#include <iostream>
#include <stdio.h>
using namespace std;

int n;
int arr[10000] = { 0 };
int sum[10000] = { 0 };
int ans = 0;

int main()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> arr[i];
		sum[i] = sum[i - 1] + arr[i];
	}
	for (int i = 1; i <= n; i++)
	{
		ans = ans + arr[i] * (sum[n] - sum[i]);
	}
	printf("%d", ans);
	return 0;
}