Road HDU - 5861 线段树

最新推荐文章于 2020-01-27 10:07:07 发布

原创最新推荐文章于 2020-01-27 10:07:07 发布 · 322 阅读

0 ·

CC 4.0 BY-SA版权

线段树|树状数组专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种利用线段树数据结构解决城市间过路费最小化的问题。通过记录门的开关时间和过路费，实现了对于每天出行计划的最优化计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意: 有n个城市在一条直线上,每两个城市之间有个门,经过这个门是有过路费的,每个门只能开一次关一次,然后给你m个计划，每天一个计划，每个计划有两个值 a,b,表示从a到b。问你每天最少的过路费；

思路：先用线段树，把每个门的开始和结束时间记录下来，然后处理一下，将开始的时间的地方+过路费，结束的时间的下一个地方-过路费

注意：先是RE ，可能是a< b。然后TLE ，在用线段树球求开始和结束时间的时候，如果只用一个lazy往下推的话，会超时。然后WA，当那个门没有开启的时候，如果算进去了，因为我们是在结束的下一个地方减去过路费。。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <vector>

using namespace std;
#define LL long long
const int maxn = 200015;

int lazy[maxn<<2],be[maxn<<2],en[maxn<<2],mon[maxn];
LL ans[maxn];
void update(int i,int l,int r)
{
    if(l!=r){
       if(be[i<<1]==0) be[i<<1]=be[i],en[i<<1]=en[i];
        else if(en[i]>en[i<<1]) en[i<<1]=en[i];
        if(be[i<<1|1]==0) be[i<<1|1]=be[i],en[i<<1|1]=en[i];
        else if(en[i]>en[i<<1|1]) en[i<<1|1]=en[i];
        lazy[i<<1]=lazy[i],lazy[i<<1|1]=lazy[i];
        be[i]=en[i]=0;
    }
    lazy[i]=0;
}
void sert(int i,int l,int r,int a,int b,int t)
{
    if(l==a&&r==b)
    {
        if(lazy[i]) update(i,l,r);
        lazy[i]=t;if(be[i]==0) be[i]=en[i]=t;else if(en[i]<t) en[i]=t;
        return ;
    }
    int mid=(l+r)>>1;
    if(lazy[i]) update(i,l,r);
    if(b<=mid) sert(i<<1,l,mid,a,b,t);
    else if(a>=(mid+1)) sert(i<<1|1,mid+1,r,a,b,t);
    else {
        sert(i<<1,l,mid,a,mid,t);
        sert(i<<1|1,mid+1,r,mid+1,b,t);
    }
}

int n,m;
int flag=0;
void updown(int i,int l,int r)
{
    if(l==r){
        if(l<=(n-1)&&be[i]){
            if(lazy[i]) update(i,l,r);
            ans[be[i]]+=mon[l],ans[en[i]+1]-=mon[l];
        }
        return ;
    }
    int mid=(l+r)>>1;
    if(lazy[i]) update(i,l,r);
    updown(i<<1,l,mid);
    updown(i<<1|1,mid+1,r);
}
void init()
{
    for(int i=0;i<=n;i++)
        mon[i]=0;
    for(int i=0;i<=(m+1);i++)
        ans[i]=0;
    memset(be,0,sizeof(be));
    memset(en,0,sizeof(en));
    memset(lazy,0,sizeof(lazy));
    return ;
}
int main()
{
    while(scanf("%d %d",&n,&m)!=EOF){
        init();
        for(int i=1;i<=n-1;i++)
            scanf("%d",&mon[i]);
        for(int i=1;i<=m;i++)
        {
            int l,r;
            scanf("%d %d",&l,&r);
            if(l>r) swap(l,r);
            sert(1,1,200000,l,r-1,i);
        }
        flag=0;
        updown(1,1,200000);
        LL total=0;
        for(int i=1;i<=m;i++)
        {
            total+=ans[i];
            printf("%I64d\n",total);
        }
    }
    return 0;
}