洛谷 P3932 浮游大陆的68号岛 前缀和

题目

https://www.luogu.org/problemnew/show/P3932
妖精仓库的储物点可以看做在一个数轴上。每一个储物点会有一些东西，同时他们之间存在距离。

每次他们会选出一个小妖精，然后剩下的人找到区间[l,r][l,r]储物点的所有东西，清点完毕之后问她，把这个区间内所有储物点的东西运到另外一个仓库的代价是多少？

比如储物点ii有xx个东西，要运到储物点jj，代价为

x \times \mathrm{dist}( i , j )x×dist(i,j)
dist就是仓库间的距离。

当然啦，由于小妖精们不会算很大的数字，因此您的答案需要对19260817取模。

题解

维护数组cnt，sum_dist，cost，表示储物点东西数量，到储物点距离，把1-i的物品都搬到i点所花费的代价 这三个前缀和

求[l,r]搬到x所花费的代价：假设l < r < x
先把所有物品搬到r上，//把1-r的物品搬到r上（cost），多花费了把1-l的物品搬到l上（cost）和把1-l的物品从l搬到r两部分（cnt*sum_dist），减去这两部分即可
再把所有物品搬到x上 //前缀和求出物品总数* r x的距离差即可

代码

#include<iostream>
#include<cstdio>
using namespace std;

const int mod=19260817;
const int N=200000+500;
long long cnt1[N],cnt2[N],cost1[N],cost2[N],sum_dist1[N],sum_dist2[N];
long long dis[N],w[N];

long long n,m,x,l,r;

void init(){
    for(int i=1;i<=n;i++){//正向 
        sum_dist1[i]=(sum_dist1[i-1]%mod+dis[i]%mod)%mod;
        cnt1[i]=(cnt1[i-1]+w[i])%mod;
        cost1[i]=(cost1[i-1]%mod+(cnt1[i-1]%mod*dis[i]%mod)%mod%mod)%mod;
    }
    for(int i=n;i>=1;i--){//反向 
        sum_dist2[i]=(sum_dist2[i+1]%mod+dis[i+1]%mod)%mod;
        cnt2[i]=(cnt2[i+1]+w[i])%mod;
        cost2[i]=(cost2[i+1]%mod+(cnt2[i+1]%mod*dis[i+1]%mod)%mod%mod)%mod;
    }
}
long long work(long long xx,long long ll,long long rr){
    long long ans=0;
    if(xx<=ll){
        ans=((cost2[ll]-cost2[rr]+mod)%mod-(cnt2[rr+1]%mod*(sum_dist2[ll]-sum_dist2[rr]+mod)%mod)%mod+mod)%mod;
        ans+=((cnt2[ll]-cnt2[rr+1]+mod)%mod*(sum_dist2[x]-sum_dist2[ll]+mod)%mod)%mod;
        ans=(ans+mod)%mod;
    }
    else if(xx>=rr){
        ans=((cost1[rr]-cost1[ll]+mod)%mod-(cnt1[ll-1]%mod*(sum_dist1[rr]-sum_dist1[ll]+mod)%mod)%mod+mod)%mod;
        ans+=((cnt1[rr]-cnt1[ll-1]+mod)%mod*(sum_dist1[x]-sum_dist1[rr]+mod)%mod)%mod;
        ans=(ans+mod)%mod;
    }
    else if(ll<x&&x<rr){
        ans=(work(x,ll,x)+work(x,x,rr))%mod;
    }
    return ans;
} 
int main(){
    scanf("%lld%lld",&n,&m);
    for(int i=2;i<=n;i++){
        scanf("%lld",&dis[i]);
    }
    for(int i=1;i<=n;i++) scanf("%lld",&w[i]);
    init();
    /*for(int i=1;i<=n;i++){
    //  printf("i=%d,cnt1=%lld,cnt2=%lld\n",i,cnt1[i],cnt2[i]);
        //printf("i=%d,dist1=%lld,dist2=%lld\n",i,sum_dist1[i],sum_dist2[i]);
        printf("i=%d,cost1=%lld,cost2=%lld\n",i,cost1[i],cost2[i]);
    }*/
    for(int i=1;i<=m;i++){
        scanf("%lld%lld%lld",&x,&l,&r);
        printf("%lld\n",work(x,l,r));
    }
    return 0;
}