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

题目传送门

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

---

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

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

Code

 #include<cstdio>
 #include<algorithm>

 using namespace std;

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

 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)>>;
     build(p*,l,mid);
     build(p*+,mid+,r);
 }

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

 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)>>;
     if(l>mid) change(p*+,l,r,x);
     else if(r<=mid) change(p*,l,r,x);
     else change(p*,l,mid,x),change(p*+,mid+,r,x);
     t[p].val=max(t[p*].val,t[p*+].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)>>;
     if(l>mid) return ask(p*+,l,r);
     else if(r<=mid) return ask(p*,l,r);
     else return max(ask(p*,l,mid),ask(p*+,mid+,r));
 }

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

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