南邮 OJ 1409 售票系统

售票系统

时间限制(普通/Java) :  1000 MS/ 3000 MS          运行内存限制 : 65536 KByte
总提交 : 252            测试通过 : 38 

比赛描述

某次列车途经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<D<=C，1<=N<=C，所有的售票申请按申请的时间从早到晚给出。

输出

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

样例输入

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

样例输出

YES
YES
NO
NO

提示

 

题目来源

JSOI2010

#include<stdio.h>
#define MAX_N 60010

struct TreeNode{
	int left;		//最左边线段编号
	int right;		//最右端线段编号
	int maxUsed;	//本跨度线段最多被使用的次数
	int childToAdd;	//本节点已经加过，但是孩子节点没有被加过，等待查询时加的使用计数
}nodes[MAX_N<<2];	//4倍容量肯定够存储

//构建以节点i为根节点的树
void build(int i, int left, int right){
	nodes[i].left = left;
	nodes[i].right = right;
	nodes[i].maxUsed = 0;
	nodes[i].childToAdd = 0;
	if(left != right){
		int mid = (left+right)>>1;
		build(i<<1|1, left, mid);
		build((i+1)<<1, mid+1, right);
	}
}

//[left,right]区间使用计数增加used,更新树
void add(int i, int left, int right, int used){
	if(nodes[i].left==left && nodes[i].right==right){
		nodes[i].maxUsed += used;
		nodes[i].childToAdd += used;		//先存着，向孩子插入时 或者 查询时更新
	}else{
		int lChild = i<<1|1;
		int rChild = (i+1)<<1;
		int temp = nodes[i].childToAdd;		//插入子节点之前，先将childToAdd加到孩子节点
		if(temp){
			nodes[lChild].maxUsed += temp;
			nodes[lChild].childToAdd += temp;
			nodes[rChild].maxUsed += temp;
			nodes[rChild].childToAdd += temp;
			nodes[i].childToAdd = 0;
		}
		int mid = (nodes[i].left+nodes[i].right)>>1;
		if(mid>=left){						//RE 需要更新左子树
			temp = right;
			if(temp>mid){
				temp = mid;
			}
			add(lChild, left, temp, used);
		}
		if(mid+1<=right){					//RE 需要更新右子数
			temp = left;
			if(temp<mid+1){
				temp = mid+1;
			}
			add(rChild, temp, right, used);
		}
		if(nodes[lChild].maxUsed<nodes[rChild].maxUsed){
			nodes[i].maxUsed = nodes[rChild].maxUsed;
		}else{								//更新完子树之后，更新当前节点最大使用次数
			nodes[i].maxUsed = nodes[lChild].maxUsed;
		}
	}
}

//查询[left,right]中最大使用次数
int maxUsed(int i, int left, int right){
	if(nodes[i].left==left && right==nodes[i].right){
		return nodes[i].maxUsed;
	}
	int lChild = i<<1|1;
	int rChild = (i+1)<<1;
	int temp = nodes[i].childToAdd;			//查询子节点之前，先将childToAdd加到孩子节点
	if(temp){
		nodes[lChild].maxUsed += temp;
		nodes[lChild].childToAdd += temp;
		nodes[rChild].maxUsed += temp;
		nodes[rChild].childToAdd += temp;
		nodes[i].childToAdd = 0;
	}
	int mid = (nodes[i].left+nodes[i].right)>>1;
	if(mid>=right){							//区间包含在左子树之内
		return maxUsed(lChild, left, right);
	}
	if(mid+1<=left){						//区间包含在右子树之内
		return maxUsed(rChild, left, right);
	}
	int result = maxUsed(lChild, left, mid);
	temp = maxUsed(rChild, mid+1, right);
	if(result<temp){
		result = temp;
	}
	return result;
}


int main(){
	int C,S,R,O,D,N;
	scanf("%d%d%d",&C,&S,&R);
	build(0,1,C-1);							//第一段路编号1，最后一段路编号C-1，共C-1段路，C个站点
	while(R--){
		scanf("%d%d%d",&O,&D,&N);
		if(maxUsed(0,O,D-1)+N<=S){
			printf("YES\n");
			add(0,O,D-1,N);
		}else{
			printf("NO\n");
		}
	}
}






/*Time Limit Exceed at Test 11
#include<stdio.h>
int a[60001];
int main(){
	int C,S,R,O,D,N,i;
	scanf("%d%d%d",&C,&S,&R);
	for(i=1;i<=C;i++){
		a[i] = S;
	}
	while(R--){
		scanf("%d%d%d",&O,&D,&N);
		for(i=O;i<D;i++){
			if(a[i]<N){
				break;
			}
		}
		if(i<D){
			printf("NO\n");
		}else{
			printf("YES\n");
			for(i=O;i<D;i++){
				a[i] -= N;
			}
		}
	}
}
*/