算法:
前缀和,开始以为要离散化一下,写到一半发现不用.
思路:
就是按着题意给的意思模拟这个过程就好了.取模的时候别偷懒,会wa的.
代码:
#include <bits/stdc++.h> using namespace std; typedef long long ll; const int mod=1e9+7; const int N=2e5+500; ll sum[N],cnt[N],pos[N],b[N]; ll Ans(ll u) { return (u%mod+mod)%mod; } ll dis(int L,int R,int u) { ll ans=Ans(pos[u]*(cnt[R]-cnt[L-1]))-Ans(sum[R]-sum[L-1]); if(R<=u) return Ans(ans); else return Ans(-ans); } int main() { int n,m,x,l,r;; scanf("%d%d",&n,&m);pos[1]=1; for(int i=2;i<=n;i++) { scanf("%d",&x);pos[i]+=pos[i-1]+x;pos[i]%=mod; } for(int i=1;i<=n;i++) scanf("%lld",&b[i]); for(int i=1;i<=n;i++) { cnt[i]+=cnt[i-1]+b[i];//统计下这个区间有多少个线段. cnt[i]%=mod; sum[i]+=sum[i-1]+b[i]*pos[i]%mod; sum[i]%=mod; } for(int i=1;i<=m;i++) { scanf("%d%d%d",&x,&l,&r); if(r<=x)//[l,r]->x. { cout<<dis(l,r,x)<<'\n'; } else if(x<=l)//x->[l,r]. { cout<<dis(l,r,x)<<'\n'; } else//[l,x]+[x,r]. { cout<<Ans(dis(l,x,x)+dis(x,r,x))<<'\n'; } } return 0; }