建设道路(前缀和优化)

在这里插入图片描述
在这里插入图片描述
思路:我觉得这个题目有点意思。。。。用一个简单的数学公式展开可以将O(n^2)降到O(n)实属牛皮。
sum1是ai的前缀和,sum2是ai ^ 2的前缀和
∑ i = 1 n \displaystyle\sum_{i=1}^{n} i=1n ∑ j = i + 1 n ( a i − a j ) 2 \displaystyle\sum_{j=i + 1}^{n} (ai - aj) ^2 j=i+1n(aiaj)2
=> ∑ i = 1 n \displaystyle\sum_{i=1}^{n} i=1n ∑ j = i + 1 n ( a i 2 − 2 ∗ a i ∗ a j + a j 2 ) \displaystyle\sum_{j=i + 1}^{n} (ai ^ 2 - 2 * ai * aj + aj ^2 ) j=i+1n(ai22aiaj+aj2)
=> ∑ i = 1 n ( n − i ) ∗ a i 2 \displaystyle\sum_{i=1}^{n}(n - i) * ai ^ 2 i=1n(ni)ai2 + ∑ j = i + 1 n a j 2 \displaystyle\sum_{j=i + 1}^{n} aj ^2 j=i+1naj2 - 2ai ∑ j = i + 1 n a j \displaystyle\sum_{j=i + 1}^{n} aj j=i+1naj
=> ∑ i = 1 n ( n − i ) ∗ a i 2 − 2 ∗ a i ∗ s u m 1 [ n ] − s u m [ i ] + s u m 2 [ n ] − s u m 2 [ i ] \displaystyle\sum_{i=1}^{n}(n - i) * ai ^ 2 -2 * ai *sum1[n] - sum[i] +sum2[n] - sum2[i] i=1n(ni)ai22aisum1[n]sum[i]+sum2[n]sum2[i]

代码:

#include<bits/stdc++.h>
using namespace std;

const int maxn = 5e5 + 10; 
typedef long long int ll;
ll mod = 1e9 + 7;
ll a[maxn];
ll sum1[maxn],sum2[maxn];
void solved(){
	int n;scanf("%d",&n);
	for(int i = 1; i <= n; i++){
		scanf("%lld",&a[i]);
		sum1[i] = (sum1[i - 1] + a[i]) % mod;
		sum2[i] = (sum2[i - 1] + a[i] * a[i] % mod) %mod; 
	}
	ll ans = 0;
	for(int i = 1; i <= n; i++){
		ans += (((n - i) * a[i] % mod * a[i] % mod) %mod - (2 * a[i] % mod * (sum1[n] - sum1[i] + mod) %mod) %mod + (sum2[n] - sum2[i] + mod)%mod)%mod; 
	}
	printf("%lld\n",ans%mod);
}
int main(){
	solved();
	return 0;
} 

感觉这样的题目还是蛮有意思的。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值