洛谷P5686 [CSP-S2019 江西] 和积和题解

最新推荐文章于 2024-07-17 20:43:45 发布

Roy吴承锦

最新推荐文章于 2024-07-17 20:43:45 发布

阅读量306

点赞数

分类专栏：洛谷题解文章标签： c++ 算法

本文链接：https://blog.csdn.net/RoyWu_091207/article/details/126371783

版权

洛谷题解专栏收录该内容

6 篇文章 0 订阅

订阅专栏

大佬们都很牛，本蒟蒻在此提供一种极其复杂的解法。

Part 1

首先，大家都能看出，前缀和可以将时间复杂度优化至 $O(n^2)$ ，预测 $70 pt s$ 。

Part 2

考虑对前缀和进行优化。

$\sum_{l = 1}^{n}\sum_{r = l}^{n}\left[(sa_r - sa_{l - 1}) \cdot (sb_r - sb_{l - 1})\right]$

把这个式子分配一下，得

$\sum_{l = 1}^{n}\sum_{r = l}^{n}[sa_r \cdot sb_r - sa_r \cdot sb_{l - 1} - sa_{l - 1} \cdot sb_r + sa_{l - 1} \cdot sb_{l - 1}]$

如果把第二层求和代入进去，得

$\sum_{l = 1}^n[\sum_{r = l}^n [sa_r \cdot sb_r] - \sum_{r = 1}^n sa_r \cdot sb_{l - 1} - \sum_{r = 1}^n sb_r \cdot sa_{l - 1} + sa_{l - 1} \cdot sb_{l - 1} \cdot (n - l + 1)]$

这就简单了。

等式第一项，处理前缀和数组 $sa_i \cdot sb_i$ 的前缀和 $msab_i$ ，单项时间复杂度优化至 $O (n)$ ;
等式第二项和第三项，分别处理前缀和数组 $sa_i,sb_i$ 的前缀和 $ssa_i,ssb_i$ ，单项时间复杂度同样优化至 $O (n)$ ；
等式第四项，时间复杂度 $O (n)$ ；
总时间复杂度 $O (n)$ ，可以 $100 pt s$ 了！

注意

为避免溢出，需要不断取模，不能放过一个细节，否则你会万劫不复……

~~（我写这道题时就是这个原因一直没过！现在过了，真的坑）~~

AC Code：

#include <iostream> 
#include <cstdio>

using namespace std;

typedef long long ll;

const ll MOD = 1e9 + 7;

ll a[5000001], b[5000001], sa[5000001], sb[5000001], ssa[5000001], ssb[5000001], msab[5000001], n; 
/*数组分别是：a数组，b数组，前缀和数组sa、sb，前缀和的前缀和数组ssa、ssb，前缀和数组的积的前缀和数组msab*/

int main() {
	cin >> n;
	for (int i = 1; i <= n; i++)
		cin >> a[i], sa[i] = (a[i] + sa[i - 1]) % MOD, ssa[i] = (sa[i] + ssa[i - 1]) % MOD;
	for (int i = 1; i <= n; i++)
		cin >> b[i], sb[i] = (b[i] + sb[i - 1]) % MOD, ssb[i] = (sb[i] + ssb[i - 1]) % MOD;
	for (int i = 1; i <= n; i++)
		msab[i] = (sa[i] * sb[i] % MOD + msab[i - 1]) % MOD;
	ll ans = 0;
	for (int i = 1; i <= n; i++)
		ans += ((((((msab[n] - msab[i - 1]) % MOD + MOD) % MOD //第一项
			   + (sa[i - 1] * sb[i - 1]) % MOD * (n - i + 1) % MOD //第四项
			   - (ssa[n] - ssa[i - 1]) * sb[i - 1]) % MOD + MOD) % MOD //第二项
			   - (ssb[n] - ssb[i - 1]) * sa[i - 1]) % MOD + MOD) % MOD; //第三项
	cout << ans % MOD;
	return 0;
}