约会安排 HDU - 4553

原题链接
考察:线段树
思路:
  如果做过校门外的树(增强版)那道题,应该可以想到是建立两个线段树.一个维护基友和女神约会时间tr[0],一个维护女神时间tr[1].
  对于基友相约t,我们求出tr[0]的t个空白时间的最左端点,然后更新tr[0];对于女神相约t,先求出tr[0]的左端点,没有就求tr[1]的左端点.更新tr[1]与tr[0].对于虚假的学习,两个都做清空操作.
  接下来就是考虑怎么求最左端点.这实际上是求出长度至少为t的空闲连续区间(最大连续子段).考虑设置Ls(从左端点出发的最大连续长度),Rs(同理).那么这个空闲区间可能在

1. 左子区间内部.
2. 左子区间右子区间中间.
3. 右子区间内部.
     这就是和那道连通村庄一样的求解方式了.
     为了方便判断是否可行,可以再维护一个变量sum,指[l,r]区间内最大的连续长度.
     此题细节超多,比如push_up与push_down
     然后就是存不存在使用的tr[s][1].sum是还未更新的.答案是不存在,我们每次进行相应操作,都会更新相应的tr[s][1],我们怎么修改基友占用时间,都不会对女神线段树造成影响.

Code

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010,M = 10;
char op[M];
int n,m;
struct Node{
	int l,r,Ls,Rs,sum,tag;//ls以左端点空白时间 ,Rs以右端点空白时间,整个区间的最大连续空白时间 
}tr[2][N<<2];//0:女神+基友 1:女神 
int get(int u)
{
	return tr[0][u].r-tr[0][u].l+1;
}
void push_up(int u,int s)
{
	tr[s][u].Ls = tr[s][u<<1].Ls+(tr[s][u<<1].Ls==get(u<<1)?tr[s][u<<1|1].Ls:0);
	tr[s][u].Rs = tr[s][u<<1|1].Rs+(tr[s][u<<1|1].Rs==get(u<<1|1)?tr[s][u<<1].Rs:0);
	tr[s][u].sum = max(tr[s][u<<1].sum,tr[s][u<<1|1].sum);
	tr[s][u].sum = max(tr[s][u<<1].Rs+tr[s][u<<1|1].Ls,tr[s][u].sum);
}
void push_down(int u,int s)
{
	if(tr[s][u].tag==1)
	{
		tr[s][u<<1].Ls = tr[s][u<<1].Rs = tr[s][u<<1].sum = get(u<<1);
		tr[s][u<<1|1].Ls = tr[s][u<<1|1].Rs = tr[s][u<<1|1].sum = get(u<<1|1);
		tr[s][u<<1].tag = tr[s][u<<1|1].tag = tr[s][u].tag;
		tr[s][u].tag = 0;
	}
	if(tr[s][u].tag==-1)
	{
		tr[s][u<<1].Ls = tr[s][u<<1].Rs = tr[s][u<<1].sum = 0;
		tr[s][u<<1|1].Ls = tr[s][u<<1|1].Rs = tr[s][u<<1|1].sum = 0;
		tr[s][u<<1].tag = tr[s][u<<1|1].tag = tr[s][u].tag;
		tr[s][u].tag = 0;
	}
}
void build(int u,int l,int r)
{
	tr[0][u] = {l,r,1,1,1,0}; tr[1][u] = {l,r,1,1,1,0};
	if(l==r) return;
	int mid = l+r>>1;
	build(u<<1,l,mid); build(u<<1|1,mid+1,r);
	push_up(u,0); push_up(u,1);
}
int findL(int u,int s,int c)//清空操作会破坏连续性,所以要维护Rs 
{
	if(tr[s][u].l==tr[s][u].r) return tr[s][u].l;
	int mid = tr[s][u].l+tr[s][u].r>>1;
	push_down(u,s);
	if(tr[s][u<<1].sum>=c) return findL(u<<1,s,c);
	else if(tr[s][u<<1].Rs+tr[s][u<<1|1].Ls>=c) return mid-tr[s][u<<1].Rs+1;
	else return findL(u<<1|1,s,c);
}
void modify(int u,int l,int r,int s)//清零 
{
	if(tr[s][u].l>=l&&tr[s][u].r<=r)
	{
		tr[s][u].Ls = tr[s][u].Rs = tr[s][u].sum = 0;
		tr[s][u].tag = -1;
		return;
	}
	push_down(u,s);
	int mid = tr[s][u].l+tr[s][u].r>>1;
	if(l<=mid) modify(u<<1,l,r,s);
	if(mid<r) modify(u<<1|1,l,r,s);
	push_up(u,s);
}
void clearT(int u,int l,int r,int s)
{
	if(tr[s][u].l>=l&&tr[s][u].r<=r)
	{
		tr[s][u].Ls = tr[s][u].Rs = tr[s][u].sum = get(u);
		tr[s][u].tag = 1;
		return;
	}
	push_down(u,s);
	int mid = tr[s][u].l+tr[s][u].r>>1;
	if(l<=mid) clearT(u<<1,l,r,s);
	if(mid<r) clearT(u<<1|1,l,r,s);
	push_up(u,s);
}
int main()
{
	int T,kcase=0;
	scanf("%d",&T);
	while(T--)
	{
		printf("Case %d:\n",++kcase);
		scanf("%d%d",&n,&m);
		build(1,1,n);
		while(m--)
		{
			int c,d;
			scanf("%s%d",op,&c);
			if(op[0]=='D')
			{
				if(tr[0][1].sum<c) {printf("fly with yourself\n");continue;} 
				int L = findL(1,0,c);
				modify(1,L,L+c-1,0);
				printf("%d,let's fly\n",L);
			}else if(op[0]=='N')
			{
				int s = -1;
				if(tr[0][1].sum>=c) s = 0;
				else if(tr[1][1].sum>=c) s = 1;
				if(s==-1) {puts("wait for me");continue;} 
				int L = findL(1,s,c);
				printf("%d,don't put my gezi\n",L);
				modify(1,L,L+c-1,1),modify(1,L,L+c-1,0);
			}else if(op[0]=='S')
			{
				scanf("%d",&d);
				puts("I am the hope of chinese chengxuyuan!!");
				clearT(1,c,d,0); clearT(1,c,d,1);
			}
		}
	}
	return 0;
}