Luogu P1607 庙会班车【线段树】By cellur925

题目传送门

据说可以用贪心做？算了算了...我都不会贪...。

---

开始想的是用线段树，先建出一颗空树，然后输进区间操作后就维护最大值，显然开始我忽视了班车的容量以及可以有多组奶牛坐在一起的信息。

我们肯定想要有更多的区间被选中，根据线段覆盖问题的套路，我们要优先选结束位置最小的，能选就选，但是需要判断下能不能选，就是比较下当前区间最大值和容量的关系，尽量加上，进行修改。说白了就是在维护带修改的区间最大值。

Code

#include<cstdio>
#include<algorithm>

using namespace std;

int k,n,c,ans;
struct moo{
    int f,t,sum;
}cow[50090];
struct SegmentTree{
    int l,r,val,lazy;
}t[20090*4];

bool cmp(moo a,moo b)
{
    return a.t<b.t;
}

void build(int p,int l,int r)
{
    t[p].l=l,t[p].r=r;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
}

void update(int p)
{
    if(!t[p].lazy) return ;
    if(t[p].l==t[p].r) return ;
    t[p*2].val+=t[p].lazy;
    t[p*2+1].val+=t[p].lazy;
    t[p*2].lazy+=t[p].lazy;
    t[p*2+1].lazy+=t[p].lazy;
    t[p].lazy=0;
}

void change(int p,int l,int r,int x)
{
    update(p);
    if(t[p].l==l&&t[p].r==r)
    {
        t[p].val+=x;
        t[p].lazy+=x;
        return ;
    }
    int mid=(t[p].l+t[p].r)>>1;
    if(l>mid) change(p*2+1,l,r,x);
    else if(r<=mid) change(p*2,l,r,x);
    else change(p*2,l,mid,x),change(p*2+1,mid+1,r,x);
    t[p].val=max(t[p*2].val,t[p*2+1].val);
}

int ask(int p,int l,int r)
{
    update(p);
    if(t[p].l==l&&t[p].r==r) return t[p].val;
    int mid=(t[p].l+t[p].r)>>1;
    if(l>mid) return ask(p*2+1,l,r);
    else if(r<=mid) return ask(p*2,l,r);
    else return max(ask(p*2,l,mid),ask(p*2+1,mid+1,r));
}

int main()
{
    scanf("%d%d%d",&k,&n,&c);
    for(int i=1;i<=k;i++)
        scanf("%d%d%d",&cow[i].f,&cow[i].t,&cow[i].sum);
    sort(cow+1,cow+1+k,cmp);
    build(1,1,n);
    for(int i=1;i<=k;i++)
    {
        int sta=ask(1,cow[i].f,cow[i].t),res=0;
        if(sta>=c) continue;
        if(sta+cow[i].sum<=c) res=cow[i].sum;
        else res=c-sta;
        ans+=res;
        change(1,cow[i].f,cow[i].t-1,res);
    }
    printf("%d",ans);
    return 0;
}

注意，最后修改时[l,r]的区间应在[l,r-1]上做改动，因为在r时刻，可理解为那群奶牛立即下车，有新的奶牛立即上车。