您的位置:首页 > 其它

hdu 5861 Road 线段树区间更新

2016-08-24 09:30 441 查看
考虑每一段最早出现的时间和最晚消失的时间,在这之间这段路都是要打开的,

然后将2*n个时间点放到m个查询,对每个查询,可知道当前查询有哪些点进入,哪些点出去,

统计它们对答案的贡献。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn=220000+5;
using namespace std;

int w[maxn],a[maxn][2];
int cnt[maxn*4];
vector<int>in[maxn],out[maxn];

void update(int k,int l,int r,int a,int b,int v){
if(r<a||b<l)return ;
if(a<=l&&r<=b){cnt[k]=v;return ;}
if(cnt[k]>0){
cnt[k<<1]=cnt[k];
cnt[k<<1|1]=cnt[k];
cnt[k]=-1;
}
int mid=(l+r)>>1;
update(k<<1,l,mid,a,b,v);
update(k<<1|1,mid+1,r,a,b,v);
}

int query(int k,int l,int r,int a){
if(l==r)return cnt[k];
if(cnt[k]>0)return cnt[k];
int mid=(l+r)>>1;
if(a<=mid)return query(k<<1,l,mid,a);
else return query(k<<1|1,mid+1,r,a);
}

int main()
{
int n,m;
while(~scanf("%d%d",&n,&m)){
for(int i=0;i<maxn;i++)in[i].clear(),out[i].clear();
memset(cnt,0,sizeof(cnt));
for(int i=1;i<n;i++){
scanf("%d",&w[i]);
}
for(int i=1;i<=m;i++){
scanf("%d%d",&a[i][0],&a[i][1]);
if(a[i][0]>a[i][1])swap(a[i][0],a[i][1]);a[i][1]--;
update(1,1,n,a[i][0],a[i][1],i);
}
for(int i=1;i<n;i++){
int tmp=query(1,1,n,i);
out[tmp].push_back(i);
}
memset(cnt,0,sizeof(cnt));
for(int i=m;i>=1;i--){
update(1,1,n,a[i][0],a[i][1],i);
}
for(int i=1;i<n;i++){
int tmp=query(1,1,n,i);
in[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;
}

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  线段树