JZOJ 1294. 轻轨

题目

Description

　　有N（1<=N<=20,000）个站点的轻轨站，有一个容量为C（1<=C<=100）的列车起点在1号站点，终点在N号站点，有K(K<=50,000)组牛群，每组数量为M_i（1<=M_i<=N），行程起点和终点分别为S_i和E_i（1<=S_i<E_i<=N）
　　计算最多有多少头牛可以搭乘轻轨。

 

Input

　　第1行：3个空格隔开的整数K,N,C
　　第2到K+1行：描述每组奶牛的情况，每行3个空格隔开的整数S_i,E_i和M_i

Output

　　输出一个整数表示最多可以搭乘轻轨的牛的数量。

 

Sample Input

8 15 3
1 5 2
13 14 1
5 8 3
8 14 2
14 15 1
9 12 1
12 15 2
4 6 1

Sample Output

10

 

Data Constraint

 

 

Hint

【样例说明】
　　轻轨从1号站搭2头牛到5号站，搭3头牛从5好站到8号站，2头牛从8号站到14号站，1头牛从9号站到12号站，1头牛从13号站到14号站，1头牛从14号站到15号站。

 

分析

• 线段树覆盖？
• 贪心？
• 区间修改，查询最大值

 

代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

struct sb
{
    int st,ed,num;
}a[200001];
struct AA
{
    int l,r,add,sum,mx;
}t[200100*4];
bool cmp(sb a,sb b)
{
    if (a.ed!=b.ed)
    return a.ed<b.ed?true:false; 
    else return a.st<b.st?true:false;
}
void build(int a,int b,int k)
{
    t[k].l=a; t[k].r=b;
    if (a==b) return;
    int mid=(a+b)/2;
    build(a,mid,k*2);
    build(mid+1,b,k*2+1);
}
void lazy(int k,int m)
{
    if (t[k].add)
    {
        t[k*2].add+=t[k].add;
        t[k*2+1].add+=t[k].add;
        t[k*2].mx+=t[k].add;
        t[k*2+1].mx+=t[k].add;
        t[k].add=0; 
    }
}
void change(int x,int y,int z,int k)
{
    if (t[k].l>=x&&t[k].r<=y)
    {
        t[k].add+=z;
        t[k].mx+=z;
        return;
    }
    lazy(k,t[k].r-t[k].l+1);
    int mid=(t[k].l+t[k].r)/2;
    if (y<=mid) change(x,y,z,k*2);
    else if (x>mid) change(x,y,z,k*2+1);
    else
    {
        change(x,mid,z,k*2); 
        change(mid+1,y,z,k*2+1);
    }
    t[k].sum=t[k*2].sum+t[k*2+1].sum;
    t[k].mx=max(t[k*2].mx,t[k*2+1].mx);
}
int check(int x,int y,int k)
{
    if (t[k].l>=x&&t[k].r<=y)
        return t[k].mx;
    lazy(k,t[k].r-t[k].l+1);
    int mid=(t[k].l+t[k].r)/2;
    int val=0;
    if (x<=mid) val=max(val,check(x,y,k*2));
    if (y>mid) val=max(val,check(x,y,k*2+1));
    return val;
}
int main()
{
    int k,n,c;
    scanf("%d%d%d",&k,&n,&c);
    for (int i=1;i<=k;i++) scanf("%d%d%d",&a[i].st,&a[i].ed,&a[i].num);
    sort(a+1,a+1+k,cmp);
    build(1,20001,1);
    int ans=0;
    for (int i=1;i<=k;i++)
    {
        int maxn=check(a[i].st,a[i].ed-1,1);
        ans+=min(a[i].num,c-maxn);
        change(a[i].st,a[i].ed-1,min(a[i].num,c-maxn),1);
    }
    cout<<ans;
    return 0;
}