【BZOJ2819】Nim 博弈論+樹鏈剖分

又一道博弈論,先來了解一下Nim遊戲~

Nim遊戲:一種經典 ICG (Impartial Combinatorial Games)——兩個人進行遊戲,N堆石子,每回合可以取其中某一堆的任意多個,可以取完,但不可以不取。誰不能取誰輸。 我們可以將每一個情況根據是否有必勝策略而分爲 N-position(先手必勝)和 P-position(後手必勝),對於一個局面,它是 P-position 當且僅當a^a₂^…^an=0

於是,這道題的代碼實現是一個樹鏈剖分的裸題。用線段樹維護區間異或和。

樹鏈剖分部分請參看【算法雜談_02】樹鏈剖分

需要注意的是題目敘述中“這三個點會導致你DFS時爆棧”。

解決辦法:

①不用DFS,而改爲BFS(代碼複雜度++)

用棧模擬DFS(代碼複雜度++)

③不要問我這是什麼。我是雷鋒~

#pragma comment(linker, "/STACK:1024000000,1024000000")
④事實上,這道題根本不用考慮DFS爆系統棧!!!因爲真的不會爆!!!

Nim C++代碼實現:

#include <cctype>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define lson pos<<1
#define rson pos<<1|1
#define N 500010
int n,m,a[N];
int cnt,head[N],next[2*N],to[2*N];
int now,size[N],deep[N],fa[N],son[N],p_id[N],id_p[N],top[N];
int xort[4*N];
void add(int x,int y)
{
	to[++cnt]=y;
	next[cnt]=head[x];
	head[x]=cnt;
}
void dfs(int x,int d)
{
	size[x]=1;deep[x]=d;
	for(int y,i=head[x];i;i=next[i])
		if((y=to[i])!=fa[x])
		{
			fa[y]=x;
			dfs(y,d+1);
			if(size[y]>size[son[x]])
				son[x]=y;
			size[x]+=size[y];
		}
}
void create(int x,int d)
{
	p_id[x]=++now;id_p[now]=x;
	top[x]=d;
	if(son[x])
		create(son[x],d);
	for(int y,i=head[x];i;i=next[i])
		if((y=to[i])!=fa[x]&&y!=son[x])
			create(y,y);
}
void build(int pos,int l,int r)
{
	if(l==r){ xort[pos]=a[id_p[l]]; return; }
	int mid=(l+r)>>1;
	build(lson,l,mid);
	build(rson,mid+1,r);
	xort[pos]=xort[lson]^xort[rson];
}
void fix(int pos,int l,int r,int x,int y)
{
	if(l==x&&r==x){ xort[pos]=y; return; }
	int mid=(l+r)>>1;
	if(x<=mid)
		fix(lson,l,mid,x,y);
	else fix(rson,mid+1,r,x,y);
	xort[pos]=xort[lson]^xort[rson];
}
int query(int pos,int l,int r,int x,int y)
{
	if(x<=l&&r<=y) return xort[pos];
	int mid=(l+r)>>1;
	if(y<=mid)
		return query(lson,l,mid,x,y);
	if(x>mid)
		return query(rson,mid+1,r,x,y);
	return query(lson,l,mid,x,y)^query(rson,mid+1,r,x,y);
}
int getnim(int x,int y)
{
	int f1=top[x],f2=top[y],re=0;
	while(f1!=f2)
	{
		if(deep[f1]<deep[f2])
			swap(x,y),swap(f1,f2);
		re^=query(1,1,n,p_id[f1],p_id[x]);
		x=fa[f1];
		f1=top[x];
	}
	if(deep[x]>deep[y])
		swap(x,y);
	return re^query(1,1,n,p_id[x],p_id[y]);
}
bool getopt()
{
	char ch;
	while(!isalpha(ch=getchar()));
	return ch=='Q';
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int x,y,i=1;i<n;i++)
		scanf("%d%d",&x,&y),
		add(x,y),add(y,x);
	dfs(1,1);
	create(1,1);
	build(1,1,n);
	cin>>m;
	for(int x,y,i=1;i<=m;i++)
	{
		bool opt=getopt();
		scanf("%d%d",&x,&y);
		if(opt)
			printf("%s",getnim(x,y)?"Yes\n":"No\n");
		else
			fix(1,1,n,p_id[x],y);
	}
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章