2819: Nim

2819: Nim

Time Limit: 20 Sec   Memory Limit: 128 MB
Submit: 2121   Solved: 823
[ Submit][ Status][ Discuss]

Description

著名游戏设计师vfleaking，最近迷上了Nim。普通的Nim游戏为：两个人进行游戏，N堆石子，每回合可以取其中某一堆的任意多个，可以取完，但不可以不取。谁不能取谁输。这个游戏是有必胜策略的。于是vfleaking决定写一个玩Nim游戏的平台来坑玩家。
为了设计漂亮一点的初始局面，vfleaking用以下方式来找灵感：拿出很多石子，把它们聚成一堆一堆的，对每一堆编号1,2,3,4,...n,在堆与堆间连边，没有自环与重边，从任意堆到任意堆都只有唯一一条路径可到达。然后他不停地进行如下操作：

1.随机选两个堆v,u，询问若在v到u间的路径上的石子堆中玩Nim游戏，是否有必胜策略，如果有，vfleaking将会考虑将这些石子堆作为初始局面之一，用来坑玩家。
2.把堆v中的石子数变为k。

由于vfleaking太懒了，他懒得自己动手了。请写个程序帮帮他吧。

Input

 第一行一个数n，表示有多少堆石子。
接下来的一行，第i个数表示第i堆里有多少石子。
接下来n-1行，每行两个数v,u，代表v,u间有一条边直接相连。
接下来一个数q，代表操作的个数。
接下来q行，每行开始有一个字符：
如果是Q，那么后面有两个数v,u，询问若在v到u间的路径上的石子堆中玩Nim游戏，是否有必胜策略。
如果是C，那么后面有两个数v,k，代表把堆v中的石子数变为k。

对于100%的数据：
1≤N≤500000, 1≤Q≤500000, 0≤任何时候每堆石子的个数≤32767
其中有30%的数据：
石子堆组成了一条链，这3个点会导致你DFS时爆栈（也许你不用DFS？）。其它的数据DFS目测不会爆。

注意：石子数的范围是0到INT_MAX

Output

对于每个Q，输出一行Yes或No，代表对询问的回答。

Sample Input

【样例输入】
5
1 3 5 2 5
1 5
3 5
2 5
1 4
6
Q 1 2
Q 3 5
C 3 7
Q 1 2
Q 2 4
Q 5 3

Sample Output

Yes
No
Yes
Yes
Yes

HINT

Source

湖北省队互测

[ Submit][ Status][ Discuss]



对于Nim游戏有个结论，初始状态所有石子xor为0则先手必败，否则先手必胜。。

那剩下的问题就简单了，维护每个点到根的xor值，求一个dfs序，线段树区间修改

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<algorithm>
using namespace std;

const int maxn = 5E5 + 50;
const int T = 4;

int n,m,dfs_clock,dfn[maxn],va[maxn],fa[maxn][20]
	,c[maxn*T],L[maxn],ot[maxn],Mark[maxn*T];
bool Huge[maxn];

vector <int> v[maxn];

void Dfs(int x,int from)
{
	dfn[x] = ++dfs_clock;
	for (int i = 1; i < 20; i++) fa[x][i] = fa[fa[x][i-1]][i-1];
	for (int i = 0; i < v[x].size(); i++)
	{
		int to = v[x][i];
		if (to == from) continue;
		L[to] = L[x] + 1; fa[to][0] = x; Dfs(to,x);
	}
	ot[x] = dfs_clock;
}

void pushdown(int o,int l,int r)
{
	if (!Mark[o]) return;
	if (l == r)
	{
		c[o] ^= Mark[o];
		Mark[o] = 0;
		return;
	}
	Mark[o<<1] ^= Mark[o];
	Mark[o<<1|1] ^= Mark[o];
	Mark[o] = 0;
}

void Modify(int o,int l,int r,int ml,int mr,int k)
{
	if (ml <= l && r <= mr)
	{
		Mark[o] ^= k;
		pushdown(o,l,r);
		return;
	}
	pushdown(o,l,r);
	int mid = (l + r) >> 1;
	if (ml <= mid) Modify(o<<1,l,mid,ml,mr,k);
	if (mr > mid) Modify(o<<1|1,mid+1,r,ml,mr,k);
}

int Query(int o,int l,int r,int pos)
{
	pushdown(o,l,r);
	if (l == r) return c[o];
	int mid = (l + r) >> 1;
	if (pos <= mid) return Query(o<<1,l,mid,pos);
	else return Query(o<<1|1,mid+1,r,pos);
}

int LCA(int p,int q)
{
	if (L[p] < L[q]) swap(p,q);
	for (int j = 18; j >= 0; j--)
		if (L[p] - (1<<j) >= L[q])
			p = fa[p][j];
	if (p == q) return p;
	for (int j = 18; j >= 0; j--)
		if (fa[p][j] != fa[q][j])
			p = fa[p][j],q = fa[q][j];
	return fa[p][0];
}

int getcom()
{
	char ch = getchar();
	while (ch != 'Q' && ch != 'C') ch = getchar();
	return ch == 'Q'?1:2;
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	cin >> n;
	for (int i = 1; i <= n; i++) scanf("%d",&va[i]);
	for (int i = 1; i < n; i++)
	{
		int x,y; scanf("%d%d",&x,&y);
		v[x].push_back(y);
		v[y].push_back(x);
	}
	L[n/2] = 1; Dfs(n/2,0); cin >> m;
	for (int i = 1; i <= n; i++) Modify(1,1,n,dfn[i],ot[i],va[i]);
	while (m--)
	{
		int com = getcom(),x,y; scanf("%d%d",&x,&y);
		if (com == 1)
		{
			int lca = LCA(x,y);
			int A = Query(1,1,n,dfn[x]);
			int B = Query(1,1,n,dfn[y]);
			if (!(A^B^va[lca])) puts("No"); else puts("Yes");
		}
		else Modify(1,1,n,dfn[x],ot[x],y^va[x]),va[x] = y;
	}
	return 0;
}