BZOJ 1095: [ZJOI2007]Hide 捉迷藏 动态树分治

1095: [ZJOI2007]Hide 捉迷藏

Time Limit: 40 Sec  Memory Limit: 256 MB
Submit: 4043  Solved: 1702
[Submit][Status][Discuss]

Description

　　捉迷藏 Jiajia和Wind是一对恩爱的夫妻，并且他们有很多孩子。某天，Jiajia、Wind和孩子们决定在家里玩捉迷藏游戏。他们的家很大且构造很奇特，由N个屋子和N-1条双向走廊组成，这N-1条走廊的分布使得任意两个屋子都互相可达。游戏是这样进行的，孩子们负责躲藏，Jiajia负责找，而Wind负责操纵这N个屋子的灯。在起初的时候，所有的灯都没有被打开。每一次，孩子们只会躲藏在没有开灯的房间中，但是为了增加刺激性，孩子们会要求打开某个房间的电灯或者关闭某个房间的电灯。为了评估某一次游戏的复杂性，Jiajia希望知道可能的最远的两个孩子的距离（即最远的两个关灯房间的距离）。 我们将以如下形式定义每一种操作： C(hange) i 改变第i个房间的照明状态，若原来打开，则关闭；若原来关闭，则打开。 G(ame) 开始一次游戏，查询最远的两个关灯房间的距离。

Input

　　第一行包含一个整数N，表示房间的个数，房间将被编号为1,2,3…N的整数。接下来N-1行每行两个整数a, b，表示房间a与房间b之间有一条走廊相连。接下来一行包含一个整数Q，表示操作次数。接着Q行，每行一个操作，如上文所示。

Output

　　对于每一个操作Game，输出一个非负整数到hide.out，表示最远的两个关灯房间的距离。若只有一个房间是关着灯的，输出0；若所有房间的灯都开着，输出-1。

Sample Input

8
1 2
2 3
3 4
3 5
3 6
6 7
6 8
7
G
C 1
G
C 2
G
C 1
G

Sample Output

4
3
3
4

HINT

对于100%的数据， N ≤100000, M ≤500000。

非常棒的题

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0;char ch=getchar();
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100,inf=0X3f3f3f3f;

struct Que
{
	priority_queue<int> A,B;
	void push(int x){if(x!=-inf)A.push(x);}
	void pop(int x){if(x!=-inf)B.push(x);}
	int top()
	{
		while(!B.empty()&&A.top()==B.top())A.pop(),B.pop();
		return !A.empty()?A.top():-inf;
	}
	int two()
	{
		if(A.size()-B.size()<2)return -inf;
		int x=top();A.pop();
		int y=top();A.push(x);
		return x+y;
	}
}q1[N],q2[N],q;

vector<int>vec[N];

int ecnt,last[N];
struct EDGE{int to,nt;}e[N<<1];
inline void add(int u,int v)
{e[++ecnt]=(EDGE){v,last[u]};last[u]=ecnt;}

int n,sum,off;
int root,size[N],f[N];

bool vis[N],black[N];

void getroot(int u,int fa)
{
	size[u]=1;f[u]=0;
	for(int i=last[u];i;i=e[i].nt)
	if(e[i].to^fa&&!vis[e[i].to])
	{
		getroot(e[i].to,u);size[u]+=size[e[i].to];
		f[u]=max(size[e[i].to],f[u]);
	}
	f[u]=max(f[u],sum-size[u]);
	if(f[u]<f[root])root=u;
}
// source destination
void seek(int u,int fa,int d,int dest)
{
	q2[dest].push(d);
	vec[u].push_back(d);
	for(int i=last[u];i;i=e[i].nt)
	if(!vis[e[i].to]&&e[i].to^fa)
	seek(e[i].to,u,d+1,dest);
}

int fa[N];

void init(int u,int ffa)
{
	fa[u]=ffa;vis[u]=1;
	q1[u].push(0);
	for(int i=last[u];i;i=e[i].nt)
	if(!vis[e[i].to])
	{
		sum=size[e[i].to];root=0;getroot(e[i].to,u);
		seek(e[i].to,u,1,root);
		q1[u].push(q2[root].top());
		init(root,u);
	}
	q.push(q1[u].two());
}

void setrev(int u)
{
	register int tmp1,tmp2,tmp3,tmp4,v,i;
	tmp1=q1[u].two();
	black[u]?q1[u].pop(0):q1[u].push(0);
	tmp2=q1[u].two();
	if(tmp1^tmp2)q.pop(tmp1),q.push(tmp2);
	for(v=u,i=vec[u].size()-1;fa[v];v=fa[v],i--)
	{
		tmp1=q2[v].top();
		black[u]?q2[v].pop(vec[u][i]):q2[v].push(vec[u][i]);
		tmp2=q2[v].top();
		if(tmp1==tmp2)continue;
		tmp3=q1[fa[v]].two();
		q1[fa[v]].pop(tmp1);q1[fa[v]].push(tmp2);
		tmp4=q1[fa[v]].two();
		if(tmp3^tmp4)q.pop(tmp3),q.push(tmp4);
	}
}

int main()
{
	n=off=sum=read();
	register int i,u,v;
	for(i=1;i<n;++i){u=read();v=read();add(u,v),add(v,u);}
	for(i=1;i<=n;++i)black[i]=1;
	f[0]=inf;getroot(1,0);init(root,0);
	int Q=read();char opt[2];
	while(Q--)
	{
		scanf("%s",opt);
		if(opt[0]=='C')
		{
			u=read();setrev(u);
			off+=black[u]?-1:1;black[u]^=1;
		}
		else
		{
			if(!off){puts("-1");}
			else if(off==1){puts("0");}
			else print(q.top()),puts("");
		}
	}
	return 0;
}
/*
8
1 2
2 3
3 4
3 5
3 6
6 7
6 8
7
G
C 1
G
C 2
G
C 1
G

4
3
3
4
*/