HDU 4553 约会安排 (超详解)

高亮查看

一. 题目

---

二. 代码

	由题知，我们需要知道是否存在一段连续区间，即线段树维护最大子段长度。
	maxv, pre, lat：最大子段长度, 最大前缀, 最大后缀，详见代码
	fa, ls, rs:父, 左儿子, 右儿子。
	fa.maxv=(ls.lat+rs.pre, max(ls.maxv, rs.maxv));
	fa.pre=ls.pre+(ls.pre==左区间长度?rs.pre:0);
	fa.lat=rs.lat+(rs.lat==右区间长度?ls.lat:0);
	此外，还知道，分配给男孩的时间不会影响女孩的时间，而分配给女孩
和学习的时间会影响男孩的时间，因此需要维护两个线段树，详见代码。
	最后利用query找出最靠前的符合要求连续子段。
	详见代码

---

三. 代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;
struct Node
{
	int l, r;
	int maxv_boy, pre_boy, lat_boy, lazy_boy;//1有, 0无 //男孩时间 
	int maxv_girl, pre_girl, lat_girl, lazy_girl;//女孩时间 
};
void build(int u, int l, int r);
void modify(int u, int l, int r, int x, int type);//0是girl and study, 1是boy,
int query(int u, int len, int type);//0是girl, 1是boy, len是需要长度, 返回起点下标 
void push_up(int u);
void push_down(int u);

int T;
int n, m;
Node tr[4*N];

signed main()
{
	cin>>T;
	
	for(int C=1;C<=T;C++)
	{
		scanf("%d%d", &n, &m);
		
		build(1, 1, n);
		
		printf("Case %d:\n", C);
		while(m--)
		{
			int a, b;
			char op[9];
			scanf("%s", op);
			
			if(*op=='D')
			{
				scanf("%d", &a);
				if(tr[1].maxv_boy<a) puts("fly with yourself");//男孩无时间 
				else
				{
					int l=query(1, a, 1);
					printf("%d,let's fly\n", l);
					modify(1, l, l+a-1, 0, 1);
				}
			}
			else if(*op=='N')
			{
				scanf("%d", &a);
				if(tr[1].maxv_boy>=a)//第一次查找 
				{
					int l=query(1, a, 1);
					printf("%d,don't put my gezi\n", l);
					modify(1, l, l+a-1, 0, 0);
				}
				else if(tr[1].maxv_girl>=a)//第二次查找 
				{
					int l=query(1, a, 0);
					printf("%d,don't put my gezi\n", l);
					modify(1, l, l+a-1, 0, 0);
				}
				else puts("wait for me");//没找到 
			}
			else//学习 
			{
				scanf("%d%d", &a, &b);
				puts("I am the hope of chinese chengxuyuan!!");
				modify(1, a, b, 1, 0);
			}
		}
	}
}

int query(int u, int len, int type)
{
	if(tr[u].l==tr[u].r) return tr[u].l;
	
	push_down(u);
	Node &fa=tr[u], &ls=tr[u<<1], &rs=tr[u<<1|1];
	int mid=fa.l+fa.r>>1;
	
	if(type)//男孩询问 
	{
		if(ls.maxv_boy>=len) return query(u<<1, len, type);//右边最大子段长度大于等于需要的 
		if(ls.lat_boy+rs.pre_boy>=len) return mid-ls.lat_boy+1;//位于左右中间, 算出该段起点 
		return query(u<<1|1, len, type);//在右边 
	}
	else
	{
		if(ls.maxv_girl>=len) return query(u<<1, len, type);//同上 
		if(ls.lat_girl+rs.pre_girl>=len) return mid-ls.lat_girl+1;
		return query(u<<1|1, len, type);
	}
}

void modify(int u, int l, int r, int x, int type)
{
	if(tr[u].l>=l&&tr[u].r<=r)
	{
		if(!type)//女孩或者学习 
		{
			tr[u].lazy_boy=tr[u].lazy_girl=x;//女孩或学习, 影响男女两个线段树 
			tr[u].maxv_boy=tr[u].pre_boy=tr[u].lat_boy=(tr[u].r-tr[u].l+1)*x;
			tr[u].maxv_girl=tr[u].pre_girl=tr[u].lat_girl=(tr[u].r-tr[u].l+1)*x;
		}
		else
		{
			tr[u].lazy_boy=x;//男孩只影响男孩的线段树 
			tr[u].maxv_boy=tr[u].pre_boy=tr[u].lat_boy=(tr[u].r-tr[u].l+1)*x;
		}
	}
	else
	{
		push_down(u);
		
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid) modify(u<<1, l, r, x, type);
		if(r>mid) modify(u<<1|1, l, r, x, type);
		
		push_up(u);
	}
}

void push_down(int u)
{
	Node &fa=tr[u], &ls=tr[u<<1], &rs=tr[u<<1|1];
	
	if(fa.lazy_boy!=-1)
	{
		ls.maxv_boy=ls.pre_boy=ls.lat_boy=(ls.r-ls.l+1)*fa.lazy_boy;//前后缀maxv, 都是新值乘区间长度 
		rs.maxv_boy=rs.pre_boy=rs.lat_boy=(rs.r-rs.l+1)*fa.lazy_boy;
		ls.lazy_boy=rs.lazy_boy=fa.lazy_boy, fa.lazy_boy=-1;//更新懒标记 
	}
	
	if(fa.lazy_girl!=-1)
	{
		ls.maxv_girl=ls.pre_girl=ls.lat_girl=(ls.r-ls.l+1)*fa.lazy_girl;
		rs.maxv_girl=rs.pre_girl=rs.lat_girl=(rs.r-rs.l+1)*fa.lazy_girl;
		ls.lazy_girl=rs.lazy_girl=fa.lazy_girl, fa.lazy_girl=-1;
	}
}

void push_up(int u)
{
	Node &fa=tr[u], &ls=tr[u<<1], &rs=tr[u<<1|1];
	fa.maxv_boy=max(ls.lat_boy+rs.pre_boy, max(ls.maxv_boy, rs.maxv_boy));//最大子段为, ls rs的最大子段 ls.lat+rs.pre 中最大的一个 
	fa.maxv_girl=max(ls.lat_girl+rs.pre_girl, max(ls.maxv_girl, rs.maxv_girl));
	fa.pre_boy=ls.pre_boy+(ls.pre_boy==ls.r-ls.l+1?rs.pre_boy:0);//最长前缀为, 左儿子的最长前缀, 当ls.pre==区间长度时, 还要加上rs.pre 
	fa.pre_girl=ls.pre_girl+(ls.pre_girl==ls.r-ls.l+1?rs.pre_girl:0);
	fa.lat_boy=rs.lat_boy+(rs.lat_boy==rs.r-rs.l+1?ls.lat_boy:0);//同前缀求法 
	fa.lat_girl=rs.lat_girl+(rs.lat_girl==rs.r-rs.l+1?ls.lat_girl:0);
}

void build(int u, int l, int r)
{
	tr[u]={l, r, 1, 1, 1, -1, 1, 1, 1, -1};
	if(l==r) return;
	
	int mid=l+r>>1;
	build(u<<1, l, mid), build(u<<1|1, mid+1, r);
	
	push_up(u);
}