【题解】 洛谷 P5686 和积和

这道题大家应该都会$O(n)$的做法了，这里讲一下$O(nlogn)$的做法吧。

首先这道题要我们求的是所有子区间的问题，我们自然而然的想到分治做法。

对于不跨过中点的，我们递归处理。

而重点就是讨论跨过重点的贡献。

这里先把式子给化简一下。

令$sum{a}_n={\sum }_{i=1}^{n}a[i]$，$sum{b}_n$同样

我们需要求的：

​ ${\sum }_{i=1}^r{\sum }_{j=i}^r(sum{a}_j-sum{a}_{i-1})\ast (sum{b}_j-sum{b}_{j-1})$

考虑一段区间：

我们先用$O(n)$的时间求出${\sum }_{i=mid+1}^{r}S(l,i)$

即求出：

绿色箭头的这段贡献。

而后我们考虑将左端点拓展到$l+1,l+2\dots \dots mid$然后分别求贡献。

我们考虑将左端点右移一位后答案会产生什么影响。

设$A_i$表示每个绿色箭头的前缀和，即$A_i=sum{a}_{mid+i}-sum{a}_l$

$B_i$定义也是同样。

设当前的答案是$ans$

容易得到ans= ${\sum }_{i=1}^{r-mid}{A}_i{B}_i$

我们将端点左移一位后，答案变成如下式子：

${\sum }_{i=1}^{r-mid}(A_i-a_l)(B_i-b_l)$

意思就是说每个前缀和都需要扣掉左端点的贡献，然后在求答案。

就是这样：

把蓝色部分的贡献扣掉，只计算红色贡献，就相当于左端点右移一位。

将括号乘开，在化简可得到：

${\sum }_{i=1}^{r-mid}{A}_i{B}_i+(r-mid)\ast a\ast b-a\ast {\sum }_{i=1}^{r-mid}{B}_i-b\ast {\sum }_{i=1}^{r-mid}{A}_i$

= $ans+(r-mid)\ast a\ast b-a\ast {\sum }_{i=1}^{r-mid}{B}_i-b\ast {\sum }_{i=1}^{r-mid}{A}_i$

过程中我们只需要用两个变量分别维护 ${\sum }_{i=1}^{r-mid}{B}_i$ 和 ${\sum }_{i=1}^{r-mid}{A}_i$ 就可以O(1)得到每次端点左移的答案。

---

Code

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

#define int long long

const int N = 5e5+100,P = 1e9+7;
int n;
int a[N],b[N];
int suma[N] , sumb[N] , Sumb[N],Suma[N];
int ans = 0;

int add(int x,int y){return x+y>P?x+y-P:x+y;}
int sub(int x,int y){return x-y<0?x-y+P:x-y;}
int mul(int x,int y){return x*1ll*y%P;}

void solve(int l,int r){
	if (l>r) return;
	int Ans = ans;
	if (l == r) {ans = (ans+a[l]*1ll*b[l]%P)%P;return;}//这个特判不能去
	if (l+1 == r){ans = add(ans,mul(a[l],b[l])); ans = add(ans,mul(a[r],b[r])); ans = add(ans,mul(add(a[l],a[r]),add(b[l],b[r])));return;}//这个特判去掉没关系
	int mid = l + r >> 1;
	int ans1 = 0;
	for (int i = mid+1; i <= r; i++)
        ans1 = add(ans1,mul(sub(suma[i],suma[l-1]),sub(sumb[i],sumb[l-1])));//O(n)计算ans
	ans = add(ans1,ans);//累加
	int summa = 0,summb = 0;
	for (int i = mid+1; i <= r; i++) summa = add(summa,sub(suma[i],suma[l-1]));
	for (int i = mid+1; i <= r; i++) summb = add(summb,sub(sumb[i],sumb[l-1]));//维护A，B的前缀和
	for (int i = l+1; i <= mid; i++){
		ans1 = add(ans1,mul(r-mid,mul(a[i-1],b[i-1])));
		ans1 = sub(ans1,mul(b[i-1],summa)); ans1 = sub(ans1,mul(a[i-1],summb));//上述公式O(1)求贡献
		summa = sub(summa,mul(a[i-1],r-mid)); summb = sub(summb,mul(b[i-1],r-mid));//左移一位，每一个前缀和都要去掉当前位的数，而一共有r-mid个前缀和
		ans = add(ans,ans1);//累加
	}
	solve(l,mid); solve(mid+1,r);//不跨中点的递归
}

signed main(){
	scanf("%lld",&n);
	for (int i = 1; i <= n; i++) scanf("%lld",&a[i]) , suma[i] = (suma[i-1]+a[i])%P;
	for (int i = 1; i <= n; i++) scanf("%lld",&b[i]) , sumb[i] = (sumb[i-1]+b[i])%P;
    // 求前缀和
	solve(1,n);
	printf("%lld\n",ans);
	return 0;
}