UPC-9558 区间权值

版权声明：本文为博主原创文章，转载请注明出处( • ̀ω•́ )✧ https://blog.csdn.net/wangws_sb/article/details/83339672

题目传送门

题目：小Bo有n个正整数a1..an，以及一个权值序列w1…wn，现在他定义
现在他想知道的值，需要你来帮帮他
你只需要输出答案对109+7取模后的值

输入

第一行一个正整数n
第二行n个正整数a1..an
第三行n个正整数w1..wn
1≤n≤3×105
1≤ai≤107
1≤wi≤107

输出

输出答案对109+7取模后的值

题解：对每一个w[i]，算一下所有包含它的区间的a的和sum。

AC代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
using namespace std;
#define io ios::sync_with_stdio(0),cin.tie(0)
#define ms(arr) memset(arr,0,sizeof(arr))
#define inf 0x3f3f3f
typedef long long ll;
const int mod=1e9+7;
const int maxn=3e5+7;
ll n;
ll a[maxn],w[maxn];
int main()
{
    io;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        a[i]=a[i]+a[i-1];
    }
    for(int i=1;i<=n;i++)
        cin>>w[i];
    ll l=1,r=n;
    ll sum=0;
    ll ans=0;
    while(l<=r)
    {
        sum=(sum+a[r]-a[l-1])%mod;
        if(l<r)
            ans=(ans+sum*w[l]+sum*w[r])%mod;
        else
            ans=(ans+sum*w[l])%mod;
        l++;r--;
    }
    cout<<ans<<endl;
    return 0;
}