【简单数学】求值

最新推荐文章于 2022-07-10 23:30:24 发布

Dy_Dream

最新推荐文章于 2022-07-10 23:30:24 发布

阅读量222

点赞数

分类专栏：数学文章标签：数学

本文链接：https://blog.csdn.net/Dy_Dream/article/details/83014493

版权

数学专栏收录该内容

4 篇文章 0 订阅

订阅专栏

【题目描述】

【输入格式】

第1 行，1 个整数N。
第2 行，N 个整数A1,A2,...,AN。

【输出格式】

输出仅1 个整数，表示所求的值

ai*aj*ak=ai*aj*(sum[n]-sum[j])(1<=i<j<k<=n)

设sum2[i]表示a[i]*sum[n]的前缀和,sum3[i]表示a[i]*sum[i]的前缀和

则有 ai*aj*(sum[n]-sum[j])=a[i]*((sum2[n]-sum2[i])-(sum3[n]-sum3[i]))

此时时间复杂度已经足以AC此题

#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
int n;
long long sum[1000005],a[1000005],sum2[1000005],sum3[1000005];
long long ans=0;
const int mod=1e9+7;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]),sum[i]=(sum[i-1]+a[i])%mod;
	for(int i=1;i<=n;i++) sum2[i]=(sum2[i-1]+a[i]*sum[n])%mod,sum3[i]=(sum3[i-1]+a[i]*sum[i])%mod;
	for(int i=1;i<=n;i++)
	{
		ans=(((ans+(a[i]*((sum2[n]-sum2[i])-(sum3[n]-sum3[i])))%mod)%mod)+mod)%mod;
	}
	ans=(ans*6)%mod;
	printf("%lld",ans);
}