P1607 [USACO09FEB]庙会班车Fair Shuttle

题目描述

Although Farmer John has no problems walking around the fair to collect prizes or see the shows, his cows are not in such good shape; a full day of walking around the fair leaves them exhausted. To help them enjoy the fair, FJ has arranged for a shuttle truck to take the cows from place to place in the fairgrounds.

FJ couldn't afford a really great shuttle, so the shuttle he rented traverses its route only once (!) and makes N (1 <= N <= 20,000) stops (conveniently numbered 1..N) along its path. A total of K (1 <= K <= 50,000) groups of cows conveniently numbered 1..K wish to use the shuttle, each of the M_i (1 <= M_i <= N) cows in group i wanting to ride from one stop S_i (1 <= S_i < E_i) to another stop E_i (S_i < E_i <= N) farther along the route.

The shuttle might not be able to pick up an entire group of cows (since it has limited capacity) but can pick up partial groups as appropriate.

Given the capacity C (1 <= C <= 100) of the shuttle truck and the descriptions of the groups of cows that want to visit various sites at the fair, determine the maximum number of cows that can ride the shuttle during the fair.

逛逛集市，兑兑奖品，看看节目对农夫约翰来说不算什么，可是他的奶牛们非常缺乏锻炼——如果要逛完一整天的集市，他们一定会筋疲力尽的。所以为了让奶牛们也能愉快地逛集市，约翰准备让奶牛们在集市上以车代步。但是，约翰木有钱，他租来的班车只能在集市上沿直线跑一次，而且只能停靠N(1 ≤N≤20000)个地点（所有地点都以1到N之间的一个数字来表示）。现在奶牛们分成K(1≤K≤50000)个小组，第i 组有Mi(1 ≤Mi≤N)头奶牛，他们希望从Si跑到Ti(1 ≤Si<Ti≤N)。

由于班车容量有限，可能载不下所有想乘车的奶牛们，此时也允许小里的一部分奶牛分开乘坐班车。约翰经过调查得知班车的容量是C(1≤C≤100)，请你帮助约翰计划一个尽可能满足更多奶牛愿望的方案。

输入输出格式

输入格式：

【输入】

第一行：包括三个整数：K，N和C，彼此用空格隔开。

第二行到K+1行：在第i+1行，将会告诉你第i组奶牛的信息：Si，Ei和Mi，彼

此用空格隔开。

输出格式：

【输出】

第一行：可以坐班车的奶牛的最大头数。

输入输出样例

输入样例#1：

输出样例#1：

说明

【样例说明】

班车可以把2头奶牛从1送到5，3头奶牛从5送到8，2头奶牛从8送到14，1头

奶牛从9送到12，1头奶牛从13送到14，1头奶牛从14送到15。

题解：贪心+线段树

按结束点排序放尽量多的奶牛。

每次要查询区间最大值和修改区间最大值需要线段树维护

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<cstring>

#define maxn 200005

using namespace std;

int k,n,m,ans;

struct Tree{

    int l,r,sum,s;

}tr[maxn<<];

struct GS{

    int x,y,c;

    bool operator < (const GS &a) const{

        return y<a.y;

    }

}a[maxn*];

void pushup(int rt){

    tr[rt].sum=max(tr[rt<<].sum,tr[rt<<|].sum);

    return;

}

void pushdown(int rt){

    if(!tr[rt].s)return;

    tr[rt<<].s+=tr[rt].s;tr[rt<<|].s+=tr[rt].s;

    tr[rt<<].sum+=tr[rt].s;tr[rt<<|].sum+=tr[rt].s;

    tr[rt].s=;

}

void build(int rt,int l,int r){

    tr[rt].l=l;tr[rt].r=r;

    if(l==r)return;

    int mid=(l+r)>>;

    build(rt<<,l,mid);

    build(rt<<|,mid+,r);

}

int query(int rt,int l,int r,int qx,int qy){

    pushdown(rt);

    if(l>=qx&&r<=qy){

        return tr[rt].sum;

    }

    int mid=(l+r)>>;

    if(qy<=mid)return query(rt<<,l,mid,qx,qy);

    else if(qx>mid)return query(rt<<|,mid+,r,qx,qy);

    else return max(query(rt<<,l,mid,qx,mid),query(rt<<|,mid+,r,mid+,qy));

}

void change(int rt,int l,int r,int qx,int qy,int z){

    pushdown(rt);

    if(l>=qx&&r<=qy){

        tr[rt].sum+=z;

        tr[rt].s+=z;

        return;

    }

    int mid=(l+r)>>;

    if(qy<=mid)change(rt<<,l,mid,qx,qy,z);

    else if(qx>mid)change(rt<<|,mid+,r,qx,qy,z);

    else change(rt<<,l,mid,qx,mid,z),change(rt<<|,mid+,r,mid+,qy,z);

    pushup(rt);

}

int main(){

    scanf("%d%d%d",&k,&n,&m);

    for(int i=;i<=k;i++)scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].c);

    sort(a+,a+k+);

    for(int i=;i<=k;i++){

        int p=min(a[i].c,m-query(,,n,a[i].x,a[i].y-));

        if(p){

            ans+=p;

            change(,,n,a[i].x,a[i].y-,p);

        }

    }

    printf("%d\n",ans);

    return ;

}