题目传送门:https://www.lydsy.com/JudgeOnline/problem.php?id=2819
首先根据SG定理,可得若每堆石子数量的异或值为0,则后手必胜,反之先手必胜。于是我们就需要维护一个支持树上单点修改,树链查询异或值的数据结构。
首先我们可以用树链剖分解决。然而……蒟蒻我比较懒(既懒得写也懒得调),所以我们可以采取一种被用烂了的黑科技:dfs序。
首先我们会发现,我们可以把查询一条链拆成查询两个端点到根的链的异或值,以及lca的权值。这样一个点的权值修改,查询时只对他的子树有贡献。所以我们可以求出树的dfs序,然后把单点修改转为区间修改,把树链查询转化为单点查询。
然后为了代码方便,我们可以差分,把区间l~r修改转为修改l、r+1,查询单点k转化为查询1~k的异或值,就能用树状数组维护了。
代码:
#include<cstdio> #include<cmath> #include<cstdlib> #include<cstring> #include<ctime> #include<algorithm> #include<queue> #include<vector> #include<map> #define ll long long #define ull unsigned long long #define max(a,b) (a>b?a:b) #define min(a,b) (a<b?a:b) #define lowbit(x) (x& -x) #define mod 1000000007 #define inf 0x3f3f3f3f #define eps 1e-18 #define maxn 500010 inline ll read() { ll tmp=0; char c=getchar(),f=1; for(;c<'0'||'9'<c;c=getchar())if(c=='-')f=-1; for(;'0'<=c&&c<='9';c=getchar())tmp=(tmp<<3)+(tmp<<1)+c-'0'; return tmp*f; } struct edge{ int to,nxt; }e[2*maxn]; int fir[maxn],pos[maxn],l[maxn],r[maxn],val[maxn],vis[maxn],dep[maxn]; int fa[maxn][20]; int bit[maxn]; int n,q,tot=0,cnt=0; char op[10]; void add(int x,int y){e[tot].to=y; e[tot].nxt=fir[x]; fir[x]=tot++;} void dfs(int now) { vis[now]=1; l[now]=pos[now]=++cnt; for(int i=fir[now];~i;i=e[i].nxt) if(!vis[e[i].to]){ fa[e[i].to][0]=now; dep[e[i].to]=dep[now]+1; dfs(e[i].to); } r[now]=cnt; } inline int getlca(int x,int y) { if(dep[x]<dep[y]){int tmp=x; x=y; y=tmp;} int k=int(log(n)/log(2)); for(int i=k;i>=0;i--) if(dep[x]-(1<<i)>=dep[y])x=fa[x][i]; if(x==y)return x; for(int i=k;i>=0;i--) if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i]; return fa[x][0]; } void change(int x,int k){if(x>n)return; for(;x<=n;x+=lowbit(x))bit[x]^=k;} int getxor(int x){int ans=0; for(;x;x-=lowbit(x))ans^=bit[x]; return ans;} int main() { memset(fir,255,sizeof(fir)); int i,j; n=read(); for(i=1;i<=n;i++)val[i]=read(); for(i=1;i<n;i++){ int x=read(),y=read(); add(x,y); add(y,x); } vis[1]=1; fa[1][0]=0; dep[1]=1; dfs(1); for(j=1;1<<j<=n;j++) for(i=1;i<=n;i++) if(fa[i][j-1])fa[i][j]=fa[fa[i][j-1]][j-1]; q=read(); for(i=1;i<=n;i++)change(l[i],val[i]),change(r[i]+1,val[i]); for(i=1;i<=q;i++){ scanf("%s",op); if(op[0]=='Q'){ int u=read(),v=read(); int tmp=getxor(pos[u])^getxor(pos[v])^val[getlca(u,v)]; if(tmp)printf("Yes\n"); else printf("No\n"); } else{ int u=read(),k=read(); change(l[u],k^val[u]); change(r[u]+1,k^val[u]); val[u]=k; } } }
虽然我用的是普通dfs,但是好像bzoj的栈空间挺大的,并不会爆。(不过还是哪天改个人工栈吧)