【问题描述】
某次列车途经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”,表示当前的售票申请被受理或不被受理。
【分析】
线段树的应用,记录区间最大值,标记下传。
1 #include <cstdlib>
2 #include <iostream>
3 #include <cstdio>
4 #include <cmath>
5 #include <cstring>
6 #include <queue>
7 #include <vector>
8 #define NEW(p) p=&mem[++size_point];p->from=0;p->to=0;p->cnt=p->lazy=0;p->left=NULL;p->right=NULL
9 const int maxn=60000*2+100;
10 using namespace std;
11 struct node
12 {
13 int from,to,cnt,lazy;
14 node *left,*right;
15 }*root,mem[maxn];
16 int c,s,r,size_point=0;
17
18 void make_tree(node *&t,int from,int to);
19 int count(node *&t,int from,int to);
20 void add(node *&t,int from,int to,int n);
21 void print(node *&t);
22
23 int main()
24 {
25 int i;
26
27 scanf("%d%d%d",&c,&s,&r);
28 make_tree(root,1,c);
29 for (i=1;i<=r;i++)
30 {
31 int u,v,n;
32 scanf("%d%d%d",&u,&v,&n);
33 v--;//注意是闭区间
34 if (count(root,u,v)+n<=s) {printf("YES\n");add(root,u,v,n);}
35 else printf("NO\n");
36 }
37 return 0;
38 }
39 void make_tree(node *&t,int from,int to)
40 {
41 NEW(t);
42 if (from>to) return;
43 if (from==to) {t->from=from;t->to=to;return;}
44 t->from=from;t->to=to;
45 make_tree(t->left,from,(from+to)/2);
46 make_tree(t->right,((from+to)/2)+1,to);
47 }
48 void add(node *&t,int from,int to,int n)
49 {
50 if (from==t->from && to==t->to)
51 {
52 t->cnt+=n;
53 t->lazy+=n;
54 return;
55 }
56 if (t->lazy!=0)
57 {
58 t->left->cnt+=t->lazy;t->right->cnt+=t->lazy;
59 t->left->lazy+=t->lazy;t->right->lazy+=t->lazy;
60 t->lazy=0;
61 }
62 int mid=(t->from+t->to)/2;
63 if (to<=mid) add(t->left,from,to,n);
64 else if (mid<from) add(t->right,from,to,n);
65 else //分段
66 {
67 add(t->left,from,mid,n);
68 add(t->right,mid+1,to,n);
69 }
70 t->cnt=max(t->left->cnt,t->right->cnt);
71 }
72 int count(node *&t,int from,int to)
73 {
74 if (from==t->from && to==t->to)
75 return t->cnt;
76 if (t->lazy!=0)
77 {
78 t->left->cnt+=t->lazy;t->right->cnt+=t->lazy;
79 t->left->lazy+=t->lazy;t->right->lazy+=t->lazy;
80 t->lazy=0;
81 }
82 int mid=(t->from+t->to)/2;
83 if (to<=mid) return count(t->left,from,to);
84 else if (mid<from) return count(t->right,from,to);
85 else //分段
86 {
87 return
88 max(count(t->left,from,mid),
89 count(t->right,mid+1,to));
90 }
91 }
92 void print(node *&t)//调试用
93 {
94 if (t==NULL) return;
95 printf("%d %d %d\n",t->from,t->to,t->cnt);
96 print(t->left);
97 print(t->right);
98 }
【cogs247】售票系统
时间: 2024-10-10 18:29:51