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); } }