比赛 Segment Tree Competition 评测结果 AAAAAAAAAAAA
题目名称 售票系统 最终得分 100
用户昵称 Respawn 运行时间 0.327 s
代码语言 C++ 内存使用 2.25 MiB
提交时间 2016-08-28 20:34:20
显示代码纯文本
#include<cstdio>
#include<cstring>
#include<iostream>
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
const int maxn=70000;
struct Node
{
	int sum,lazy;
};
Node a[maxn<<2];
int n,zuo,m;
void Insert(int,int,int,int,int,int);
void Update(int);
int _Max(int,int);
int Sum(int,int,int,int,int);
void Init();
int main()
{
	freopen("railway.in","r",stdin);
	freopen("railway.out","w",stdout);
	Init();
	return 0;
}
void Init()
{
	memset(a,0,sizeof(a));
	scanf("%d%d%d",&n,&zuo,&m);
	for(int i=1;i<=m;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		int g=Sum(x,y-1,1,1,n);
		if((g+z)>zuo)
		{
			printf("NO\n");
			continue;
		}
		else
		{
			printf("YES\n");//printf("adwdaad");
			Insert(x,y-1,z,1,1,n);
			
			continue;
		}
	}
}
void Insert(int s,int t,int z,int rt,int l,int r)
{
	if(s<=l&&t>=r)
	{
		a[rt].sum+=z;
		a[rt].lazy+=z;
		return;
	}
	//Update(rt);
	int mid=(l+r)>>1;
	if(s<=mid)
	{
		Insert(s,t,z,lson);
	}
	if(t>mid)
	{
		Insert(s,t,z,rson);
	}
	a[rt].sum=_Max(a[rt<<1].sum,a[rt<<1|1].sum);
}
int Sum(int s,int t,int rt,int l,int r)
{
	if(s<=l&&t>=r)
	{
		return a[rt].sum;
	}
	Update(rt);
	int mid=(l+r)>>1;
	if(t<=mid)
	{
		return Sum(s,t,lson);
	}
	if(s>mid)
	{
		return Sum(s,t,rson);
	}
	return _Max(Sum(s,t,lson),Sum(s,t,rson));
}
void Update(int rt)
{
	int x=a[rt].lazy;
	a[rt<<1].sum+=x;
	a[rt<<1].lazy+=x;
	a[rt<<1|1].sum+=x;	
	a[rt<<1|1].lazy+=x;
	a[rt].lazy=0;
}
int _Max(int a,int b)
{
	if(a>b)
	{
		return a;
	}
	return b;
}