HDU5861(线段树好题)

题解:找到每一段路最开始的时间和结束的时间,然后一起统计。区间最值问题,首选线段树。

可以先对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;
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值