【Bzoj2819】Nim

题意

给定一棵树，两个操作：
1.询问若在v到u间的路径上的石子堆中玩Nim游戏，是否有必胜策略。
2.把堆v中的石子数变为k。

---

作为一个解[ka]决[guo]了这道题的人，来分[kou]析[hu]一下算法，其实正解是Dfs序之后树状数组，但是因为懒就直接写了树剖，每个区间维护区间的异或值。因为异或的性质a^b^c=a^(b^c)。然后Nim游戏有必胜策略，当且仅当所有堆的石子异或起来不等于0，之后就可以用线段树直接解决。

---

#include <cstdio>
#include <algorithm>

#define Rep(i,s,t) for(int i=s;i<=t;i++)
#define For(i,s,t) for(int i=s;i;i=t)

using namespace std;

const int maxx = 500000 + 25;

int head[maxx],nxt[maxx<<1],to[maxx<<1];
int top[maxx],son[maxx],ftr[maxx],size[maxx],dpt[maxx],rnk[maxx];
int T[maxx<<2],a[maxx];

int n,m,x,y,z,cnt,num;
char f[8];

namespace Y{

    void Ins(int x,int y){to[++num]=y;nxt[num]=head[x];head[x]=num;}

    void Dfs1(int x){
        size[x] = 1;
        For(i , head[x] , nxt[i]){
            int now = to[i];if(now == ftr[x]) continue;
            dpt[now] = dpt[x] + 1;ftr[now] = x;
            Dfs1(now);size[x] += size[now];
            if(size[now] > size[son[x]]) son[x] = now;
        }
    }

    void Dfs2(int x,int brn){
        rnk[x] = ++cnt;top[x] = brn;
        if(son[x]) Dfs2(son[x],brn);
        For(i , head[x] , nxt[i])
            if(to[i] != ftr[x] && to[i] != son[x])
                Dfs2(to[i],to[i]); 
    }

    void modify(int i,int pos,int l,int r,int k){
        if(l == r) {T[i] = k;return;} int mid = (l+r) >> 1;
        if(pos <= mid) modify(i<<1,pos,l,mid,k);
        if(pos >  mid) modify(i<<1|1,pos,mid+1,r,k);
        T[i] = T[i<<1] ^ T[i<<1|1];
    }

    int Query(int i,int x,int y,int l,int r){
        if(x <= l && r <= y) return T[i];
        int ans = 0;int mid = (l+r) >> 1;
        if(x <= mid) ans ^= Query(i<<1,x,y,l,mid);
        if(y >  mid) ans ^= Query(i<<1|1,x,y,mid+1,r);
        return ans;
    }

    int Get(int x,int y){
        int ans = 0;
        while(top[x] != top[y]){
            if(dpt[top[x]] > dpt[top[y]]) x^=y^=x^=y;
            ans ^= Query(1,rnk[top[y]],rnk[y],1,n);
            y = ftr[top[y]];
        }
        if(rnk[x] > rnk[y]) x^=y^=x^=y;
        ans ^= Query(1,rnk[x],rnk[y],1,n);
        return ans;
    }

}

using namespace Y;

int main(){
    scanf("%d",&n);
    Rep(i , 1 , n) scanf("%d",&a[i]);
    Rep(i , 1 , n-1) scanf("%d%d",&x,&y),Ins(x,y),Ins(y,x);
    Dfs1(1);Dfs2(1,1);
    scanf("%d",&m);
    Rep(i , 1 , n) modify(1,rnk[i],1,n,a[i]);

    while( m-- ){
        scanf("%s",f);
        if(f[0] == 'Q')
            scanf("%d%d",&x,&y),puts((Get(x,y))? "Yes":"No");
        if(f[0] == 'C') scanf("%d%d",&x,&y),modify(1,rnk[x],1,n,y);
    }
    return 0;
}