售票系统

题目

描述 Description
某次列车途经C个城市，城市编号依次为1到C，列车上共有S个座位，铁路局规定售出的车票只能是坐票， 即车上所有的旅客都有座。售票系统是由计算机执行的，每一个售票申请包含三个参数，分别用O、D、N表示，O为起始站，D为目的地站，N为车票张数。售票 系统对该售票申请作出受理或不受理的决定，只有在从O到D的区段内列车上都有N个或N个以上的空座位时该售票申请才被受理。请你写一个程序，实现这个自动 售票系统。
输入格式 Input Format
第一行包含三个用空格隔开的整数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，所有的售票申请按申请的时间从早到晚给出。
输出格式 Output Format
输出共有R行，每行输出一个“YES”或“NO”，表示当前的售票申请被受理或不被受理。

样例输入
4 6 4
1 4 2
1 3 2
2 4 3
1 2 3
样例输出
YES
YES
NO
NO

题解

线段树的模板题
区间修改+查询最大值
单点修改+查询最大值

code

#include <bits/stdc++.h>
const int maxn = 6e4 + 7;
const int inf = 0x3f3f3f3f;
typedef long long LL;
using namespace std;

inline int max(int aa, int bb) { return aa > bb ? aa : bb; }
inline int min(int aa, int bb) { return aa < bb ? aa : bb; }

template <typename T>
inline void read(T &s) {
    s = 0;
    T w = 1, ch = getchar();
    while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); }
    while (isdigit(ch))  { s = (s << 1) + (s << 3) + (ch ^ 48); ch = getchar(); }
    s *= w;
}

template<typename T>
inline void write(T s) {
	if (s < 0) putchar('-'), s = -s;
	if (s > 9) write(s / 10);
	putchar(s % 10 + '0');
}

int c, n, q;
int a[maxn];
struct tree {
	int l, r, add, dat;
	#define l(x) t[x].l
	#define r(x) t[x].r
	#define add(x) t[x].add
	#define dat(x) t[x].dat
} t[maxn << 2];

void build(int p, int l, int r) {
	l(p) = l, r(p) = r;
	if (l == r) {
		dat(p) = a[l]; return ;
	}
	int mid = (l + r) >> 1;
	build(p << 1, l, mid);
	build(p << 1 | 1, mid + 1, r);
	dat(p) = dat(p << 1) + dat(p << 1 | 1);
}

void update(int p) {
	if (add(p)) {
		dat(p << 1) += add(p);
		dat(p << 1 | 1) += add(p);
		add(p << 1) += add(p);
		add(p << 1 | 1) += add(p);
		add(p) = 0;
	}
	return ;
} 

void change(int p, int l, int r, int d) {
	if (l <= l(p) && r >= r(p)) {
		dat(p) += d;
		add(p) += d;
		return ;
	}
	update(p);
	int mid = (l(p) + r(p)) >> 1;
	if (l <= mid) change(p << 1, l, r, d);
	if (r > mid) change(p << 1 | 1, l, r, d);
	dat(p) = max(dat(p << 1), dat(p << 1 | 1));
}

int query(int p, int l, int r) {
	if (l <= l(p) && r >= r(p)) return dat(p);
	update(p);
	int mid = (l(p) + r(p)) >> 1;
	int val = -(1 << 30);
	if (l <= mid) val = max(val, query(p << 1, l, r));
	if (r > mid) val = max(val, query(p << 1 | 1, l, r));
	return val;
}

int main() {
//	freopen("in.txt","r",stdin);
	read(c), read(n), read(q);
	build(1, 1, c);
	while (q--) {
		int left, right, all;
		read(left), read(right), read(all); right--;
		int tot = query(1, left, right);
		if (n - tot >= all) {
			change(1, left, right, all); puts("YES");
		}
		else puts("NO");
	}
	return 0;
}