【描述】 在成都的一条街道上,一共有 N 户人家,每个家庭有 Xi 个人,他们和谐的生活在 一起,作为全国和谐街道,他们经常会小范围组织活动,每次活动会选择一户作为聚点, 并要求某些家庭参加,为了方便通知,村长每次邀请位置连续的家庭。因为每户人数不 同,每个家庭之间有一定距离,村长希望你计算出每次邀请的家庭的移动代价。第 i 个家 庭移动到家庭 j的代价是: Xi*dis(i,j) dis(i,j)表示 i到 j的距离,村长一共安排了 m次聚会,每次邀请[Li,Ri]的家庭参加 【输入】 第一行两个数表示 n,m 第二行 n-1 个数,第 i个数表示第 i 个家庭与第 i+1 个家庭的距离 Di
第三行 n 个数,表示每个家庭的人数 Xi
之后 m行每行三个数 x l r,表示查询要把区间 [l,r]的家庭移动到 x 点的代价和
【输出】
对于每个询问输出一个数表示答案,对 19260817取模
【输入样例】
5 5
2 3 4 5
1 2 3 4 5
1 1 5
3 1 5
2 3 3
3 3 3
1 5 5
【输出样例】
125
72
9
0
70
【子任务】
对于 30%的数据, n,m≤1000
对于另外 20%的数据,所有家庭间的距离都为 1
对于另外 20%的数据,所有家庭人数都为 1
对于 100%的数据 , n,m≤200000;Xi,Di <=2*10^9
分析
(l--r)类问题
对于静态区间 ---- 前缀和 O(1)
对于动态区间 ---- 线段树或者树状数组 O(logn)
这道题显然就是前缀和啊
LL pre_d[N];//dis前缀和
LL pre_m[N];//pre_d * X 前缀和
LL pre_x[N];//x的前缀和
分类讨论一下
l,r 全在x左边
l,r 一左一右
l,r都在右边
代码很容易写出
#include<bits/stdc++.h>
#define N 200050
#define mod 19260817
#define LL long long
using namespace std;
LL pre_d[N];//dis前缀和
LL pre_m[N];//pre_d * X 前缀和
LL pre_x[N];//x的前缀和
LL n,m;
LL read(){
LL cnt=0;char ch=0;
while(!isdigit(ch))ch=getchar();
while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
return cnt;
}
int main(){
//freopen("party.in","r",stdin);
//freopen("party.out","w",stdout);
n=read(),m=read();
for(int i=2;i<=n;i++)
pre_d[i]=(pre_d[i-1]+read())%mod;
for(int i=1;i<=n;i++){
LL x=read();
pre_x[i]=(pre_x[i-1]+x)%mod;
pre_m[i]=(pre_m[i-1]+(x*pre_d[i])%mod)%mod;
}
for(int i=1;i<=m;i++){
LL x=read(),l=read(),r=read();
if(l>r) swap(l,r);
if(l>=x){//全部向右移
LL x1=pre_m[r]-pre_m[l-1];
LL x2=pre_d[x]*(pre_x[r]-pre_x[l-1]);
x1=(x1+mod)%mod;
x2=(x2+mod)%mod;
printf("%lld\n",((x1-x2)%mod+mod)%mod);
}
else
if(l<x&&r>x){// l-x x+1-r
LL x1=pre_d[x]*(pre_x[x]-pre_x[l-1]);
LL x2=pre_m[x]-pre_m[l-1];//ans1=x1-x2
x1=(x1+mod)%mod;
x2=(x2+mod)%mod;
LL x3=pre_m[r]-pre_m[x];
LL x4=pre_d[x]*(pre_x[r]-pre_x[x]);//ans2=x3-x4
x3=(x3+mod)%mod;
x4=(x4+mod)%mod;
printf("%lld\n",((x1-x2+x3-x4)%mod+mod)%mod);
}
else
if(r<=x){
LL x1=pre_d[x]*(pre_x[r]-pre_x[l-1]);
LL x2=pre_m[r]-pre_m[l-1];
x1=(x1+mod)%mod;
x2=(x2+mod)%mod;
printf("%lld\n",((x1-x2)%mod+mod)%mod);
}
}
return 0;
}
不知道大家有没有注意到
+mod)%mod
我开始写的cout<<(x1-x2)%mod
但在x1 x2都取模的情况下 x1-x2可能小于0
然后就爆0了
吃一堑长一智
老师说高手都要写一个
取模函数
int Mod(int x){
return (x+mod)%mod;
}
Remember it