区间权值

区间权值

https://ac.nowcoder.com/acm/problem/19798

题意 : 给出2个序列,一个是a序列,一个是w序列。 定义:f(l,r)=区间[l,r] ai*Wr−l+1
让你求的是L:[1-r] R:[L-r] 所有f(l,r)求和

分析: 一开始由于没看n的大小,暴力维护所有区间的值,前缀和来统计。但是n=3e5的话,复杂度就是 (n+1)*n/2
妥妥T了。

通过手写前几项,可以发现一定规律
当n=3 ans= (a1+a2+a3)w1+(a1+2a2+a3)w2+(a1+a2+a3)w3
当n=4 ans= (a1+a2+a3+a4)w1+(a1+2a2+2a3+a4)w2+(a1+2a2+2a3+a4)w3+(a1+a2+a3+a4)w4
当n=5 ans= (a1+a2+a3+a4+a5)w1+(a1+2a2+2a3+2a4+a5)w2+(a1+2a2+3a3+2a4+a5)w3+(a1+2a2+2a3+2a4+a5)w4+(a1+a2+a3+a4+a5)*w5

所以很显然是一个前缀和维护,O(n)可以解决,不要忘记取mod

AC代码:

​#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <vector>
#include <map>
#include <queue>
#include <deque>
#include <set>
#include <stack>
#include <cctype>
#include <cmath>
#include <cassert>

#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define INF 0x3f3f3f3f
#define PI 3.14159265358979323846

using namespace std;

typedef long long ll;

const ll mod=1000000007;
ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}

template<class T>inline void read(T &res)
{
    char c;T flag=1;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
    while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}

const int N=3e5+100;

ll n;

ll a[N];
ll w[N];

ll pre[N];

int main()
{
    scanf("%lld",&n);
    for(ll i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        pre[i]=(pre[i-1]+a[i])%mod;
    }
    for(ll i=1;i<=n;i++)
    {
        scanf("%lld",&w[i]);
        w[i]%=mod;
    }
    ll ans=0;
    ll sum=0;
    for(int i=1;i<=n;i++)
    {
        sum=sum+(pre[n-i+1]-pre[i-1]+mod)%mod;
        sum%=mod;
        ans=ans+(sum*w[i])%mod;
        ans%=mod;
    }
    printf("%lld\n",ans%mod);
    return 0;
}
全部评论

相关推荐

01-28 16:12
中南大学 Java
几年前还没有chatgpt的时候,刷题真的是很痛苦。刷不出来只能看题解,题解有几个问题:第一个是每次看的写题解的人都不一样,很难有一个统一的思路;第二个也是最重要的是,题解只提供了作者自己的思路,但是没有办法告诉你你的思路哪里错了。其实很少有错误的思路,我只是需要被引导到正确的思路上面去。所以传统题解学习起来非常困难,每次做不出来难受,找题解更难受。但是现在chatgpt能做很多!它可以这样帮助你&nbsp;-1.&nbsp;可以直接按照你喜欢的语言生成各种解法的题解和分析复杂度。2.&nbsp;把题和你写的代码都发给它,它可以告诉你&nbsp;你的思路到底哪里有问题。有时候我发现我和题解非常接近,只是有一点点🤏想错了。只要改这一点点就是最优解。信心倍增。3.&nbsp;如果遇到不懂的题解可以一行一行询问为什么要这样写,chatgpt不会嫌你烦。有时候我觉得自己的range写错了,其实那样写也没错,只是chat老师的题解有一点优化,这个它都会讲清楚。4.&nbsp;它可以帮你找可以用同类型解法来做的题。然后它可以保持解法思路不变,用一个思路爽刷一个类型的题。如果题目之间思路又有变化,它会告诉你只有哪里变了,其他的地方还是老思路。5.&nbsp;它也可以直接帮你总结模板,易错点。经过chat老师的指导,我最大的改变是敢刷题了。之前刷题需要先找某一个人写的算法题repo,然后跟着某一个人他的思路刷他给的几个题。如果想写别的题,套用思路失败了,没有他的题解,也不知道到底哪里错了;看别人的题解,思路又乱了。这个问题在二分查找和dp类型的题里面特别常见。但是现在有chat老师,他会针对我的代码告诉我我哪里想错了,应该怎么做;还按照我写代码的习惯帮我总结了一套属于我的刷题模板。每天写题全是正反馈!
牛客981:不刷才是爽
AI时代的工作 VS 传...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务