集市班车

题目描述

逛逛集市，兑兑奖品，看看节目对农夫约翰来说不算什么，可是他的奶牛们非常缺乏锻炼——如果要逛完一整天的集市，他们一定会筋疲力尽的。所以为了让奶牛们也能愉快地逛集市，约翰准备让奶牛们在集市上以车代步。但是，约翰木有钱，他租来的班车只能在集市上沿直线跑一次，而且只能停靠N个地点（所有地点都以1到N之间的一个数字来表示）。现在奶牛们分成K 个小组，第 i 组有Mi (1 ≤ Mi ≤N)头奶牛，他们希望从Si跑到Ti (1 ≤ Si < Ti ≤N)。由于班车容量有限，可能载不下所有想乘车的奶牛们，此时也允许小组里的一部分奶牛分开乘坐班车。约翰经过调查得知班车的容量是C，请你帮助约翰计划一个尽可能满足更多奶牛愿望的方案。
输入 第一行：包括三个整数：K，N和C，彼此用空格隔开。 第二行到K + 1行：在第i+ 1行，将会告诉你第i组奶牛的信息：Si，Ei和Mi，彼此用空格隔开。
输出 可以坐班车的奶牛的最大头数
对于100%的数据，N∈[1,10000],K∈[1,50000],C∈[1,100];

这是一道典型的想法+数据结构维护
这题与看电视相近，类似于有多台电视，问最多能看多少电视频道
所以果断先按尾排
我们优先将结束时间早的先弄上车
但怎么判断这段时间内还能上多上牛
区间查询，区间更新
线段树就可以解决了

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
#define FOR(i,x,y) for(int i=(x);i<(y);i++)
#define M 50000
#define N 10000
//复杂度 50000*log(10000); 
struct node{
    int si,ti,mi;
    bool operator <(const node &s)const{
        return ti<s.ti;
    }
}cow[M];
struct Segment{
    struct leave{
        int l,r,mx,add;
    }tree[N<<2];
    void up(int p){
        tree[p].mx=max(tree[p<<1].mx,tree[p<<1|1].mx);
    }
    void down(int p){
        if(tree[p].add==0)return;
        tree[p<<1].add+=tree[p].add;
        tree[p<<1].mx+=tree[p].add;
        tree[p<<1|1].add+=tree[p].add;
        tree[p<<1|1].mx+=tree[p].add;
        tree[p].add=0;
    }
    void build(int l,int r,int p){
        tree[p].l=l,tree[p].r=r,tree[p].mx=0,tree[p].add=0;
        if(l==r)return;
        int mid=(l+r)>>1;
        build(l,mid,p<<1);
        build(mid+1,r,p<<1|1);
        up(p);
    }
    void update(int l,int r,int a,int p){
        if(tree[p].l==l&&tree[p].r==r){
            tree[p].add+=a;
            tree[p].mx+=a;
            return;
        }
        down(p);
        int mid=(tree[p].l+tree[p].r)>>1;
        if(mid>=r)update(l,r,a,p<<1);
        else if(mid<l)update(l,r,a,p<<1|1);
        else {
            update(l,mid,a,p<<1);
            update(mid+1,r,a,p<<1|1);
        }
        up(p);
    }
    int query(int l,int r,int p){
        if(tree[p].l==l&&tree[p].r==r){
            return tree[p].mx;
        }
        down(p);
        int mid=(tree[p].l+tree[p].r)>>1;
        if(mid>=r)return query(l,r,p<<1);
        else if(mid<l)return query(l,r,p<<1|1);
        else return max(query(l,mid,p<<1),query(mid+1,r,p<<1|1));
    }
}Tree;
int main(){
    int k,n,c,ans=0;
    cin>>k>>n>>c;
    FOR(i,0,k)scanf("%d%d%d",&cow[i].si,&cow[i].ti,&cow[i].mi);
    sort(cow,cow+k);
    Tree.build(1,n,1);
    FOR(i,0,k){
        int mx=Tree.query(cow[i].si,cow[i].ti,1);
        mx=max(c-mx,0);
        mx=min(mx,cow[i].mi);
        if(mx)Tree.update(cow[i].si,cow[i].ti-1,mx,1);
        ans+=mx;
    }
    printf("%d\n",ans);
    return 0;
}