上海市计算机学会竞赛平台YACS 差分求和

上海市计算机学会竞赛平台 | YACS (iai.sh.cn)

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=500010;
int n,a[N],b[N];
signed main(){
	scanf("%lld",&n);
	for(int i=1;i<=n;i++){scanf("%lld",&a[i]);b[i]=b[i-1]+a[i];}
	if(n==1){cout<<a[1];return 0;}
	if(n==2){cout<<a[2]-a[1];return 0;}
	int k=a[2]-a[1];
	for(int i=3;i<=n;i++)k-=(i-1)*a[i]-b[i-1];
	cout<<k;
	return 0;
}