[海豹海边爆]牛客14683

原题链接:登录—专业IT笔试面试备考平台_牛客网

解题思路:首先计算出物品数量,距离,花费这三者的前缀和,就可以快速的计算出区间内的物品,x到点1的距离和区间内物品运输到点1的花费。

然后,如果要运输的点在区间左侧,那么真实花费是区间内的物品运输到点1的花费,减去区间内物品数从x点运输到点1的花费。

而如果x在区间右侧,那么真实花费就是区间内物品数从点1运输到x位置的花费,减去区间内物品运输到点1的花费。因为这里的运输花费与方向无关。

如果x在区间中,只需要把区间分成两个部分,分别在x左侧和x右侧,然后按照上面的流程处理即可。

AC代码:

#include<bits/stdc++.h>
using namespace std;
long long a[2000010],b[2000010],c[2000010],v[2000010];
//a记录所在点上物体数量,b为a的前缀和,c为花费的前缀和,v为距离的前缀和
long long n,m;
long long p=1000000007;
long long la(long long l,long long r,long long x){//x在区间左侧时,区间内物体搬运到x的花费
    return ((c[r]-c[l-1]+p)%p-(((b[r]-b[l-1]+p)%p)*v[x])%p+p)%p;
}
long long ra(long long l,long long r,long long x){//x在区间右侧时,区间内物体搬运到x的花费
    return ((((b[r]-b[l-1]+p)%p)*v[x])%p-(c[r]-c[l-1]+p)%p+p)%p;
}
int main(){
    cin>>n>>m;
    for(int i=2;i<=n;i++){//距离输入及距离前缀和计算
        long long buffer;
        scanf("%lld",&buffer);
        buffer%=p;
        v[i]=v[i-1]+buffer;
        v[i]%=p;
    }
    for(int i=1;i<=n;i++){//物品数量输入及花费前缀和计算
        scanf("%lld",&a[i]);
        a[i]%=p;
        b[i]=b[i-1]+a[i];
        c[i]=c[i-1]+a[i]*v[i];
        b[i]%=p,c[i]%=p;
    }
    while(m--){//查询输入及答案输出
        long long ans=0;
        long long l,r,x;
        scanf("%lld%lld%lld",&x,&l,&r);
        if(x<=l) ans=la(l,r,x)%p;
        else if(x>=r) ans=ra(l,r,x)%p;
        else ans=(ra(l,x-1,x)+la(x+1,r,x))%p;//如果x在中间,则分成左段查询向右移动和右段向左移动
        cout<<ans<<endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值