售票系统

售票系统

时间限制: 1 Sec 内存限制: 128 MB

题目描述

某次列车途经C个城市，城市编号依次为1到C，列车上共有S个座位，铁路局规定售出的车票只能是坐票，即车上所有的旅客都有座，售票系统是由计算机执行的，每一个售票申请包含三个参数，分别用O、D、N表示，O为起始站，D为目的地站，N为车票张数，售票系统对该售票申请作出受理或不受理的决定，只有在从O到D的区段内列车上都有N个或N个以上的空座位时该售票申请才被受理，请你写一个程序，实现这个自动售票系统。

输入

第一行包含三个用空格隔开的整数C、S和R，其中1<=C<=60000，1<=S<=60000，1<=R<=60000，C为城市个数，S为列车上的座位数，R为所有售票申请总数。接下来的R行每行为一个售票申请，用三个由空格隔开的整数O,D和N表示，O为起始站，D为目的地站，N为车票张数，其中1<=0

输出

共有R行，每行输出一个“YES”或“NO”，表示当前的售票申请被受理或不被受理。

样例输入

4 6 4
1 4 2
1 3 2
2 4 3
1 2 3

样例输出

YES
YES
NO
NO

题解

• 注意所给区间为左闭右开区间

var
 w:array[0..300005,1..4]of longint;
 i,j,k:longint;
 n,m,s:longint;
 a,b,c:longint;
function max(a,b:longint):longint;
begin
 if a>b
 then exit(a)
 else exit(b);
end;

procedure build(a,l,r:longint);
var mid:longint;
begin
 w[a,1]:=l; w[a,2]:=r; w[a,3]:=0; w[a,4]:=0;
 if l=r then exit;
 mid:=(l+r) div 2;
 build(a*2,l,mid); build(a*2+1,mid+1,r);
end;

procedure pushdown(a:longint);
begin
 inc(w[a,3],w[a,4]);
 inc(w[a*2,4],w[a,4]);
 inc(w[a*2+1,4],w[a,4]);
 w[a,4]:=0;
end;

function query(a,l,r:longint):longint;
var mid:longint;
begin
 if w[a,4]<>0 then pushdown(a);
 if (w[a,1]=l)and(w[a,2]=r) then exit(w[a,3]);
 mid:=(w[a,1]+w[a,2]) div 2;
 if r<=mid then exit(query(a*2,l,r)) else
 if l>=mid+1 then exit(query(a*2+1,l,r))
 else exit(max(query(a*2,l,mid),query(a*2+1,mid+1,r)));
end;

procedure pushup(a:longint);
var b:longint;
begin
 {if a=0 then exit;
 w[a,3]:=max(w[a*2,3],w[a*2+1,3]);}
 b:=a div 2;
 if w[b,3]<w[a,3]
 then w[b,3]:=w[a,3]
 else exit;
 a:=a div 2;
 pushup(a);
end;

procedure update(a,l,r,c:longint);
var mid:longint;
begin
 if (w[a,1]=l)and(w[a,2]=r)
 then begin inc(w[a,4],c); pushdown(a); pushup(a); exit; end;
 mid:=(w[a,1]+w[a,2]) div 2;
 if r<=mid then update(a*2,l,r,c) else
 if l>mid then update(a*2+1,l,r,c)
 else begin update(a*2,l,mid,c); update(a*2+1,mid+1,r,c); end;
end;

begin
 readln(n,s,m);
 build(1,1,n);
 for i:=1 to m do
  begin
   readln(a,b,c);
   if query(1,a,b-1)+c>s
   then writeln('NO')
   else begin writeln('YES'); update(1,a,b-1,c); end;
  end;
end.