BZOJ 2819 Nim 树链剖分

在树上链改+链查Nim游戏是否必胜。

如果是在链上的话直接xor一遍不等于0就好了。

树上的话树链剖分一下就好。

反正bzoj评测机是linux栈大233。

不过既然是xor（好像没什么关系），不如树上链查转化成点到根的链查。

用树状数组维护一下就好了。

#include <cstdio>
#include <algorithm>
using namespace std;
#define FOR(i,j,k) for(i=j;i<=k;i++)
#define FORD(i,j,k) for(i=j;i>=k;i--)
int read() {
    int s = 0, f = 1; char ch = getchar();
    for (; ch < '0' || ch > '9'; ch = getchar()) if (ch == '-') f = -1;
    for (; '0' <= ch && ch <= '9'; ch = getchar()) s = s * 10 + ch - '0';
    return s * f;
}
const int N = 500001, M = 2 * N;
int next[M], to[M], head[N], cnt = 0, c[N], w[N], dep[N], fa[N][20], pos[N], end[N], id = 0, n;

void add(int x, int y) {
    next[++cnt] = head[x]; head[x] = cnt; to[cnt] = y;
    next[++cnt] = head[y]; head[y] = cnt; to[cnt] = x;
}

void modify(int x, int v) {
    for (; x <= n; x += x & -x) c[x] ^= v;
}

int query(int x) {
    int ans = 0;
    for (; x >= 1; x -= x & -x) ans ^= c[x];
    return ans;
}

void dfs(int x) {
    int i;
    FOR(i,1,19) if(dep[x] >= (1 << i))
        fa[x][i] = fa[fa[x][i - 1]][i - 1];
    pos[x] = ++id;
    for (i = head[x]; i; i = next[i])
        if (to[i] != fa[x][0]) {
            fa[to[i]][0] = x;
            dep[to[i]] = dep[x] + 1;
            dfs(to[i]);
        }
    end[x] = id;
}

int lca(int x, int y) {
    if (dep[x] < dep[y]) swap(x, y);
    int t = dep[x] - dep[y], i;
    FOR(i,0,19) if(t & (1 << i)) x = fa[x][i];
    FORD(i,19,0) if(fa[x][i] != fa[y][i])
        x = fa[x][i], y = fa[y][i];
    return x == y ? x : fa[x][0];
}

int main() {
    static int w[N]; static char ch[2];
    n = read();
    int i, q, x, y;
    FOR(i,1,n) w[i]=read();
    FOR(i,2,n) add(read(), read());
    dfs(1);
    FOR(i,1,n) modify(pos[i], w[i]),modify(end[i] + 1, w[i]);
    q = read();
    while (q--) {
        scanf("%s", ch); x = read(); y = read();
        if (ch[0] == 'Q') {
            puts((query(pos[x]) ^ query(pos[y]) ^ w[lca(x, y)])?"Yes":"No");
        } else {
            modify(pos[x], w[x] ^ y); modify(end[x] + 1, w[x] ^ y);
            w[x] = y;
        }
    }
    return 0;
}

2819: Nim

Time Limit: 20 Sec   Memory Limit: 128 MB
Submit: 1376   Solved: 523
[ 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