1. 程式人生 > >Newcoder 39 B.儲物點的距離(水~)

Newcoder 39 B.儲物點的距離(水~)

Description

一個數軸,每一個儲物點會有一些東西,同時它們之間存在距離。

每次給個區間 [ l , r ] [l,r] ,查詢把這個區間內所有儲物點的東西運到另外一個儲物點的代價是多少?

比如儲物點 i i x x 個東西,要運到儲物點 j j

,代價為 x d i s t ( i , j
) x\cdot dist( i , j )

d i s t dist 就是儲物點間的距離。

Input

第一行兩個數表示 n , m n,m

第二行 n 1 n-1 個數,第 i i 個數表示第 i i 個儲物點與第 i + 1 i+1 個儲物點的距離 a i a_i

第三行 n n 個數,表示每個儲物點的東西個數 b i b_i

之後 m m 行每行三個數 x   l   r x\ l\ r

表示查詢要把區間 [ l , r ] [l,r] 儲物點的物品全部運到儲物點 x x 的花費

每次查詢獨立

( n , m 2 1 0 5 , 0 a i , b i 2 1 0 9 ) (n,m\le 2\cdot 10^5,0\le a_i,b_i\le 2\cdot 10^9)

Output

對於每個詢問輸出一個數表示答案

答案對 1000000007 1000000007 取模

Sample Input

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

Sample Output

125
72
9
0
70

Solution

x 1 = 0 , x i = x i 1 + a i x_1=0,x_i=x_{i-1}+a_i 表示每個儲物點的位置,把區間 [ l , r ] [l,r] 的東西全部移動到 x x ,相當於把區間 [ m i n ( l , x ) , m i n ( r , x ) ] [min(l,x),min(r,x)] 的東西右移到 x x ,把區間 [ m a x ( l , x ) , m a x ( r , x ) ] [max(l,x),max(r,x)] 的東西左移到 x x ,以左移為例,直接考慮把區間 [ l , r ] [l,r] 的物品移動到 k l k\le l 位置的代價,即為
a n s = i = l r b i ( x i x k ) = i = l r a i b i x k i = l r b i ans=\sum\limits_{i=l}^rb_i\cdot (x_i-x_k)=\sum\limits_{i=l}^ra_i\cdot b_i-x_k\cdot \sum\limits_{i=l}^rb_i
故維護一下 a i b i a_i\cdot b_i b i b_i 的字首和即可 O ( 1 ) O(1) 查詢

Code

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=200005;
#define mod 1000000007
int add(int x,int y)
{
	x+=y;
	if(x>=mod)x-=mod;
	return x;
}
int mul(int x,int y)
{
	ll z=1ll*x*y;
	return z-z/mod*mod;
}
int n,m,a[maxn],b[maxn],sum[maxn],res[maxn];
int main()
{
	scanf("%d%d",&n,&m);
	a[1]=0;
	for(int i=2;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		a[i]=add(x%mod,a[i-1]);
	}
	for(int i=1;i<=n;i++)scanf("%d",&b[i]),b[i]%=mod;
	for(int i=1;i<=n;i++)sum[i]=add(sum[i-1],mul(a[i],b[i]));
	for(int i=1;i<=n;i++)res[i]=add(res[i-1],b[i]);
	while(m--)
	{
		int l,r,x;
		scanf("%d%d%d",&x,&l,&r);
		int ll=max(l,x),rr=max(r,x);
		int ans=add(add(sum[rr],mod-sum[ll-1]),mod-mul(add(res[rr],mod-res[ll-1]),a[x]));
		ll=min(l,x),rr=min(r,x);
		ans=add(ans,add(mul(add(res[rr],mod-res[ll-1]),a[x]),add(sum[ll-1],mod-sum[rr])));
		printf("%d\n",ans);
	}
	return 0;
}