Description
n个村庄编号1~n,i到i+1村庄有一段铁路,每段铁路只能被开启一次关闭一次,铁路开启则每天需要一个花费wi,现在给出m天每天需要开启的铁路区间,问每天的最小花费,初始时所有铁路处于关闭状态
Input
多组用例,每组用例首先输入两个整数n和m,之后n-1个整数表示每段铁路开启一天的花费wi,最后m行每行两个整数l,r,表示当天l村庄和r村庄之间的铁路需要处于开启状态(1<=n,m<=200000,1<=wi<=1000,1<=l,r<=n,l!=r)
Output
输出m个整数表示这m天的最小花费
Sample Input
4 3
1 2 3
1 3
3 4
2 4
Sample Output
3
5
5
Solution
铁路能不开就不开,能关掉就尽早关掉,问题其实转化成对n-1个数进行m次区间更新最值,最小值表示该段铁路最早被使用的时间,最大值表示该段铁路最后一次被使用的时间,花费就是给这段时间每天都加上这段铁路的花费,更新用线段树,求花费用前缀和优化,总时间复杂度O(mlogn)
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 222222
#define ls (t<<1)
#define rs ((t<<1)|1)
int Max[maxn<<2],Min[maxn<<2],lazyx[maxn<<2],lazyn[maxn<<2];
void push_up(int t)
{
Max[t]=max(Max[ls],Max[rs]);
Min[t]=min(Min[ls],Min[rs]);
}
void push_down(int t)
{
if(lazyx[t]!=0)
{
int temp=lazyx[t];
lazyx[t]=0;
Max[ls]=max(Max[ls],temp),lazyx[ls]=max(lazyx[ls],temp);
Max[rs]=max(Max[rs],temp),lazyx[rs]=max(lazyx[rs],temp);
}
if(lazyn[t]!=INF)
{
int temp=lazyn[t];
lazyn[t]=INF;
Min[ls]=min(Min[ls],temp),lazyn[ls]=min(lazyn[ls],temp);
Min[rs]=min(Min[rs],temp),lazyn[rs]=min(lazyn[rs],temp);
}
}
void build(int l,int r,int t)
{
Max[t]=0,Min[t]=INF,lazyx[t]=0,lazyn[t]=INF;
if(l==r)return ;
int mid=(l+r)>>1;
build(l,mid,ls),build(mid+1,r,rs);
}
void update(int L,int R,int l,int r,int t,int v)
{
if(L<=l&&r<=R)
{
Max[t]=max(Max[t],v),lazyx[t]=max(lazyx[t],v);
Min[t]=min(Min[t],v),lazyn[t]=min(lazyn[t],v);
return ;
}
push_down(t);
int mid=(l+r)>>1;
if(L<=mid)update(L,R,l,mid,ls,v);
if(R>mid)update(L,R,mid+1,r,rs,v);
push_up(t);
}
int query_max(int L,int R,int l,int r,int t)
{
if(l==r)return Max[t];
if(L<=l&&r<=R)return Max[t];
push_down(t);
int mid=(l+r)>>1,ans=0;
if(L<=mid)ans=max(ans,query_max(L,R,l,mid,ls));
if(R>mid)ans=max(ans,query_max(L,R,mid+1,r,rs));
return ans;
}
int query_min(int L,int R,int l,int r,int t)
{
if(l==r)return Min[t];
if(L<=l&&r<=R)return Min[t];
push_down(t);
int mid=(l+r)>>1,ans=INF;
if(L<=mid)ans=min(ans,query_min(L,R,l,mid,ls));
if(R>mid)ans=min(ans,query_min(L,R,mid+1,r,rs));
return ans;
}
int n,m,w[maxn];
ll sum[maxn];
int main()
{
while(~scanf("%d%d",&n,&m))
{
memset(sum,0,sizeof(sum));
for(int i=1;i<n;i++)scanf("%d",&w[i]);
build(1,n,1);
for(int i=1;i<=m;i++)
{
int l,r;
scanf("%d%d",&l,&r);
if(l>r)swap(l,r);
r--;
update(l,r,1,n,1,i);
}
for(int i=1;i<=n;i++)
{
int tmin=query_min(i,i,1,n,1),tmax=query_max(i,i,1,n,1);
if(tmin==INF)continue;
sum[tmin]+=w[i],sum[tmax+1]-=w[i];
}
for(int i=1;i<=m;i++)
{
sum[i]+=sum[i-1];
printf("%I64d\n",sum[i]);
}
}
return 0;
}