题解:找到每一段路最开始的时间和结束的时间,然后一起统计。区间最值问题,首选线段树。
可以先对m条边倒着更新一遍线段树,便可以知道没一个点最开始的时候在哪一条线段上。然后再正着跟新一遍,就知道每个点最终结束在哪一条线段。然后合并统计
#include<bits/stdc++.h>
using namespace std;
int node[200010*4];
vector<int>in[200010],out[200010];
void update(int p,int q,int l,int r,int n,int x)
{
if(p<=l&&q>=r){node[n]=x;return;}
if(node[n]!=-1){node[n<<1]=node[n<<1|1]=node[n];node[n]=-1;}
int mid=(l+r)>>1;
if(p>mid)update(p,q,mid+1,r,n<<1|1,x);
else if(q<=mid)update(p,q,l,mid,n<<1,x);
else{update(p,q,mid+1,r,n<<1|1,x);update(p,q,l,mid,n<<1,x);}
}
int query(int l,int r,int n,int x)
{
if(l==r) return node[n];
if(node[n]!=-1) return node[n];
int mid=(l+r)>>1;
if(x<=mid)return query(l,mid,n<<1,x);
else return query(mid+1,r,n<<1|1,x);
}
int w[200010];int l[200010],r[200010];
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=0;i<=m;i++){in[i].clear();out[i].clear();}
memset(node,-1,sizeof(node));
for(int i=1;i<n;i++)scanf("%d",&w[i]);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&l[i],&r[i]);
if(l[i]>r[i]) swap(l[i],r[i]);
}
for(int i=m;i>=1;i--)update(l[i],r[i]-1,1,n,1,i);
for(int i=1;i<n;i++)
{
int tmp=query(1,n,1,i);
if(tmp!=-1)in[tmp].push_back(i);
}
memset(node,-1,sizeof(node));
for(int i=1;i<=m;i++)update(l[i],r[i]-1,1,n,1,i);
for(int i=1;i<n;i++)
{
int tmp=query(1,n,1,i);
if(tmp!=-1)out[tmp].push_back(i);
}
int ans=0;
for(int i=1;i<=m;i++)
{
for(int j=0;j<in[i].size();j++)ans+=w[in[i][j]];
printf("%d\n",ans);
for(int j=0;j<out[i].size();j++)ans-=w[out[i][j]];
}
}
return 0;
}