灵茶八题 - 子数组 +w+

灵茶八题 - 子数组 +w+

题目描述

给你一个长为 $n$ 的数组 $a$，输出它的所有连续子数组的元素和的元素和。

例如 $a=[1,2]$ 有三个连续子数组 $[1],[2],[1,2]$，元素和分别为 $1,2,3$，所以答案为 $1+2+3=6$。

输入格式

第一行输入一个整数 $n(1\le n\le 2\cdot 10^5)$。

第二行输入 $n$ 个整数，表示数组 $a$ 中的元素 $(-10^3\le a[i]\le 10^3)$。

输出格式

一个整数，表示 $a$ 的所有连续子数组的元素和的元素和。

样例 #1

样例输入 #1

2
1 2

样例输出 #1

6

样例 #2

样例输入 #2

1
1

样例输出 #2

1

提示

我们可以知道只有加号， 因此我们可以算出每一个数的贡献， 每一个数出现的次数为 i * (n - i + 1) ,贡献乘上a[i]即可；

AC代码：

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int, int>PII;
const int N=2e5+10;
const int MOD = 998244353;
const int INF=0X3F3F3F3F;
const int dx[]={-1,1,0,0,-1,-1,+1,+1};
const int dy[]={0,0,-1,1,-1,+1,-1,+1};
const int M = 1e7 + 10;

int t;

int a[N];
int main()
{
	int n;
	cin >> n;
	for(int i = 1; i <= n; i ++)
	{
		cin >> a[i];
	}
	//由于只有加法因此我们可以算每一个数的贡献，每一个数出现的次数为i * (n - i + 1);
	ll res = 0;
	for(ll i = 1; i <= n; i ++)
	{
		res += i * (n - i + 1) * a[i];
	}
	cout << res << endl;
	return 0;
}