原题链接:登录—专业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;
}