https://ac.nowcoder.com/acm/problem/14683
题意:大致意思就是把坐标轴一个区间里的所有数(权值)移动到某一点上,移动消耗为到目标点的距离*数字权值,计算总消耗。
思路:0 <= ai,bi <= 2000000000,区间数据太大,枚举显然会TLE,线段树可能会爆内存(没试过),O(1)计算区间总和,显然用前缀和比较简单。
具体实现:
1:当x<=l时 区间[l,r]到x的代价 等于区间[l,r]到1的总代价减去[l,r]的数从1到x的代价。
2:当x>=r时 区间[l,r]到x的代价 等于区间[l,r]到n的总代价减去[l,r]的数从n到x的代价。
3:其余情况,区间[l,r]到x的代价分成俩份[l,x-1]到x和[x+1,r]到x,分别求再相加。
4:记得longlong和负数取模问题就可以了^_^复杂度O(2n+m)
代码如下:
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
using namespace std;
typedef long long ll;
#define Mod 1000000007
ll n,i,j,m,x,y,l,r;
ll s[200050],p[200050],ps[200050];//s记录距离前缀和,p记录个数前缀和,ps记录价格前缀和
int main()
{
scanf("%lld%lld",&n,&m);
for(i=1;i<=n-1;i++){
scanf("%lld",&y);
s[i+1]=(s[i]+y)%Mod;//前缀和处理
}
for(i=1;i<=n;i++){
scanf("%lld",&y);
p[i]=(p[i-1]+y)%Mod;//前缀和处理
ps[i]=((y%Mod*s[i]%Mod)%Mod+ps[i-1])%Mod;//前缀和处理
}
while(m--){
ll ans=0;
scanf("%lld%lld%lld",&x,&l,&r);
if(x<=l){
ans=(ps[r]-ps[l-1]+Mod)%Mod-s[x]*(p[r]-p[l-1]+Mod)%Mod;//注意负数Mod!
}else if(x>=r){
ans=s[x]*(p[r]-p[l-1]+Mod)%Mod-(ps[r]-ps[l-1]+Mod)%Mod;//把上式取反
}else{
ll xl,xr;
xl=s[x]*(p[x]-p[l-1]+Mod)%Mod-(ps[x]-ps[l-1]+Mod)%Mod;//r变x
xr=((ps[r]-ps[x-1]+Mod)%Mod-s[x]*(p[r]-p[x-1]+Mod)%Mod);//l变x
ans+=(xl+xr)%Mod;
}
printf("%lld\n",(ans+Mod)%Mod);
}
}
京公网安备 11010502036488号