【bzoj2819】Nim

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

湖北省队互测

题解
熟练剖分裸题

代码

#include<algorithm>
#include<cstring>
#include<iostream>
#include<cstdio>
#include<cmath>
#define oo 100000000
#define ll long long
#define mod 1000000
#define N 500005
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int Head[N],ret[N*2],Next[N*2];
int size[N],dep[N],fa[N],son[N];
int bl[N],pos[N],t[N],v[N],a[N];
int tot,tim,n;
void ins(int u,int v){ret[++tot]=v;Next[tot]=Head[u];Head[u]=tot;}
void dfs1(int u)
{
    size[u]=1;
    for (int i=Head[u];i;i=Next[i])
    {
        int v=ret[i];
        if (v!=fa[u])
        {
            fa[v]=u;dep[v]=dep[u]+1;
            dfs1(v);
            size[u]+=size[v];
            if (son[u]==0||size[son[u]]<size[v]) son[u]=v;
        }
    }
}
void dfs2(int u,int chain)
{
    tim++;
    bl[u]=chain;
    pos[u]=tim;
    a[tim]=v[u];
    if (son[u]) dfs2(son[u],chain);
    for (int i=Head[u];i;i=Next[i])
    {
        int v=ret[i];
        if (v!=son[u]&&v!=fa[u]) dfs2(v,v);
    }
}
void build(int k,int l,int r)
{
    if (l==r){t[k]=a[l];return;}
    int mid=(l+r)>>1;
    build(k<<1,l,mid);build(k<<1|1,mid+1,r);
    t[k]=t[k<<1]^t[k<<1|1];
}
void modify(int k,int l,int r,int x,int y)
{
    if (l==r){t[k]=y;return;}
    int mid=(l+r)>>1;
    if (x<=mid) modify(k<<1,l,mid,x,y);
    else modify(k<<1|1,mid+1,r,x,y);
    t[k]=t[k<<1]^t[k<<1|1];
}
int query(int k,int l,int r,int x,int y)
{
    if (l==x&&r==y) return t[k];
    int mid=(l+r)>>1;
    if (y<=mid) return query(k<<1,l,mid,x,y);
    else if (x>mid) return query(k<<1|1,mid+1,r,x,y);
    else return query(k<<1,l,mid,x,mid)^query(k<<1|1,mid+1,r,mid+1,y);
}
int queryxor(int x,int y)
{
    int ans=0;
    while (bl[x]!=bl[y])
    {
        if (dep[bl[x]]<dep[bl[y]]) swap(x,y);
        ans^=query(1,1,n,pos[bl[x]],pos[x]);
        x=fa[bl[x]];
    }
    if (pos[x]>pos[y]) swap(x,y);
    ans^=query(1,1,n,pos[x],pos[y]);
    return ans;
}
int main()
{
    n=read();
    for (int i=1;i<=n;i++) v[i]=read();
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read();
        ins(u,v);ins(v,u);
    }
    dep[1]=1;
    dfs1(1);dfs2(1,1);
    build(1,1,n);
    int T=read();
    while (T--)
    {
        char ch[5];
        scanf("%s",ch);
        if (ch[0]=='Q')
        {
            int x=read(),y=read();
            if (queryxor(x,y)) puts("Yes");else puts("No");
        }
        else
        {
            int x=read(),y=read();
            modify(1,1,n,pos[x],y);
        }
    }
    return 0;
}