建设道路

链接：https://ac.nowcoder.com/acm/contest/5158/J
来源：牛客网

牛牛国有 nn 个城市，编号为 1-n1−n，第 ii 个城市有一个价值 a_ia
i
​
，牛国的国王牛阔落特别喜欢在牛牛国旅游，并且他不想每次旅游的时候都计算一遍走哪条路最短，于是他决定在任意两个城市之间建立一条双向道路，在第 ii 座城市和第 jj 座城市之间建立双向道路的代价是 (a_i-a_j)^2(a
i
​
−a
j
​
)
2
，牛阔落希望你能算出这项工程的花费。由于答案太大，你只需要输出答案模 1e9+71e9+7 的余数

输入描述:
第一行一个整数 nn，表示城市的数量。

第二行 nn 以空格分隔的整数 a1,a2,…,ana1,a2,…,an，表示第i座城市的价值。

输出描述:
输出一行一个数字，表示工程的花费模 1e9+71e9+7 的余数
示例1
输入
复制
3
1 2 3
输出
复制
6
说明
城市1到城市2的道路价值是(2 - 1)^ 2 = 1

城市2到城市3的道路价值是(3 - 2)^ 2 = 1

城市1到城市3的道路价值是(3 - 1)^ 2 = 4
总的花费 = 1 + 1 + 4 = 6
备注:
1\leq n\leq 5e51≤n≤5e5
1\leq a_i\leq 1e91≤a
i
​
≤1e9
建议使用 scanf 读入

思路：
过读题我们很容易看出他就是让我们求这么一个东西：

\sum_{i = 1}^n \sum_{j = i + 1}^{n} (a_i - a_j)^2∑
i=1
n
​
∑
j=i+1
n
​
(a
i
​
−a
j
​
)
2

我们把后边的完全平方公式展开就是:

\sum_{i = 1}^n \sum_{j = i + 1}^{n} a_i^2 -2a_ia_J + a_j^2∑
i=1
n
​
∑
j=i+1
n
​
a
i
2
​
−2a
i
​
a
J
​
+a
j
2
​

显然我们还可以把 a_i^2a
i
2
​
提出来

\sum_{i = 1}^n (n - i)a_i^2-2a_i\sum_{j = i + 1}^{n}a_J + \sum_{j = i + 1}^{n}a_j2∑
i=1
n
​
(n−i)a
i
2
​
−2a
i
​
∑
j=i+1
n
​
a
J
​
+∑
j=i+1
n
​
a
j
2
​

显然后边的a_ja
j
​
和a_j^2a
j
2
​
我们可以提前用前缀和处理

我们用sum[i]表示对a[i]数组的前缀和，用sum2表示对a[i]^2a[i]
2
的前缀和

那么原来的式子就能化成:

\sum_{i = 1}^n (n - i)a_i^2-2a_i*(sum[n] - sum[i])+ (sum2[n]-sum2[i])∑
i=1
n
​
(n−i)a
i
2
​
−2a
i
​
∗(sum[n]−sum[i])+(sum2[n]−sum2[i])

最后的时候注意取膜就行了

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 5e5 + 10;

typedef long long LL;

LL sum1[N], sum2[N], sum[N];
LL a[N];

const int mod = 1e9 + 7;

int main(){
	int n;
	scanf("%d", &n);
	for (int i = 1; i <= n; i ++) {
		scanf("%lld", &a[i]);
		sum1[i] = sum1[i - 1] % mod + (1ll * a[i] * a[i]) % mod;
		sum1[i] %= mod;
		sum2[i] = sum2[i - 1] % mod + (1ll * a[i]) % mod;
		sum2[i] %= mod;
		sum[i] = (1ll * a[i] * a[i]) % mod;  
	}
	
	LL ans = 0;
	
	for (int i = 1; i <= n; i ++){
		ans = ans % mod + (n - i) * (sum[i]) % mod;
		//	cout << i << "------" << ans << endl;
		ans = ans % mod + (sum1[n] - sum1[i]) % mod;
		//	cout << i << "------" << ans << endl;
		ans = ans % mod - (2ll * a[i] % mod * (sum2[n] - sum2[i])) % mod;
		ans = (ans % mod + mod) % mod;
		//cout << i << "------" << ans << endl;
	}
	
	cout << ans << endl;
	
	return 0;
}