牛客·区间权值（前缀和）

数学不好，被这道题虐了好久。。。

记录这悲惨的经历。

一开始，发现直接枚举f(l,r)的所有区间复杂度至少为O(n^2)一定超时，想着化简公式。

n分别取3,4,5，找到一点w[i]系数的规律，但没往前缀和上靠，搞了好久没弄出来。😂

借鉴别人的思路，才知道前缀和还能这样用！！！

#include<bits/stdc++.h> 
using namespace std;
#define int long long 
const int p=1e9+7;
const int N=3e5+10;
int a[N],w[N],s[N],t[N];
signed main()
{   ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		s[i]=(a[i]+s[i-1])%p;
		t[i]=(s[i]+t[i-1])%p;
	}
	for(int i=1;i<=n;i++)
	cin>>w[i];
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		ans=(ans+w[i]*(p+t[n]-t[i-1]-t[n-i])%p)%p;
	}
	cout<<ans;
}

注意在ans=(ans+w[i]*(p+t[n]-t[i-1]-t[n-i])%p)%p处，里面的括号内加p,防止取模取到负数了。

2024.7.19