【cogs247】售票系统【线段树】

售票系统

输入文件：railway.in 输出文件：railway.out 
时间限制：1 s 内存限制：128 MB 
【问题描述】 
某次列车途经C个城市，城市编号依次为1到C，列车上共有S个座位，铁路局规定售出的车票只能是坐票， 即车上所有的旅客都有座。售票系统是由计算机执行的，每一个售票申请包含三个参数，分别用O、D、N表示，O为起始站，D为目的地站，N为车票张数。售票 系统对该售票申请作出受理或不受理的决定，只有在从O到D的区段内列车上都有N个或N个以上的空座位时该售票申请才被受理。请你写一个程序，实现这个自动 售票系统。 
【输入格式】 
第一行包含三个用空格隔开的整数C、S和R，其中1≤C≤60000， l≤S≤60000，1≤R≤60000。C为城市个数，S为列车上的座位数，R为所有售票申请总数。接下来的R行每行为一个售票申请，用三个由空格隔开的整数O，D和N表示，O为起始站，D 为目的地站，，N为车票站数，其中1≤D≤C，1≤O≤C，所有的售票申请按申请的时间从早到晚给出。 
【输出格式】 
输出共有R行，每行输出一个“YES”或“NO”，表示当前的售票申请被受理或不被受理。 
【输入输出样例】 
Sample Input: 
4 6 4 
1 4 2 
1 3 2 
2 4 3 
1 2 3 
Sample Output: 
YES 
YES 
NO 
NO

线段树+lazy标记 
代码如下：

#include <iostream>
#include <cstdio>
using namespace std;
struct SegTreeNode {
    int lc, rc, lazy, sum;
} SegTree[600001 * 4];
int C, S, R;
void Build(int root, int l, int r)
{
    SegTree[root].lc = l, SegTree[root].rc = r;
    if (l == r)
    {
        SegTree[root].sum = 0;
        return;
    }
    int mid = (l + r) / 2;
    Build(root * 2, l, mid);
    Build(root * 2 + 1, mid + 1, r);
    SegTree[root].sum = max(SegTree[root * 2].sum, SegTree[root * 2 + 1].sum);
}
void pushdown(int root)
{
    int lc = root * 2, rc = root * 2 + 1;
    SegTree[lc].lazy += SegTree[root].lazy;
    SegTree[rc].lazy += SegTree[root].lazy;
    SegTree[lc].sum += SegTree[root].lazy;
    SegTree[rc].sum += SegTree[root].lazy;
    SegTree[root].lazy = 0;
}
void Modify(int root, int l, int r, int val)
{
    if (l == SegTree[root].lc && r == SegTree[root].rc)
    {
        SegTree[root].lazy += val;
        SegTree[root].sum += val;
        return;
    }
    pushdown(root);
    int mid = (SegTree[root].lc + SegTree[root].rc) / 2;
    if (r <= mid) Modify(root * 2, l, r, val);
    else if (l > mid) Modify(root * 2 + 1, l, r, val);
    else Modify(root * 2, l, mid, val), Modify(root * 2 + 1, mid + 1, r, val);
    SegTree[root].sum = max(SegTree[root * 2].sum, SegTree[root * 2 + 1].sum);
}
int Query(int root, int l, int r)
{
    if (l == SegTree[root].lc && r == SegTree[root].rc) return SegTree[root].sum;
    pushdown(root);
    int mid = (SegTree[root].lc + SegTree[root].rc) / 2;
    if (r <= mid) return Query(root * 2, l, r);
    else if (l > mid) return Query(root * 2 + 1, l, r);
    else
    {
        int p1 = Query(root * 2, l, mid), p2 = Query(root * 2 + 1, mid + 1, r);
        return max(p1, p2);
    }
}
int main()
{
    ios::sync_with_stdio(false);
    freopen("railway.in", "r", stdin);
    freopen("railway.out", "w", stdout);
    cin >> C >> S >> R;
    Build(1, 1, C);
    for (int i = 1; i <= R; i++)
    {
        int O, D, N, res=INT_MAX;
        cin >> O >> D >> N;
        D--;//这里一定要减一
        res = Query(1, O, D);
        if (res <= S - N) Modify(1, O, D, N);
        if (res <= S - N) cout << "YES" << endl;
        else cout << "NO" << endl;
    }
    return 0;
}