bzoj 3729: Gty的游戏 splay+dfs序+阶梯博弈

题意

给定一颗树，初始n个结点，1为根节点。每个结点上有一定的石子数。
现在你需要在线兹瓷三种操作：
1、询问以x为根的子树中进行组合游戏，双方轮流操作，每次操作可以将一个结点（在子树内且不为x）的不超过p个至少1个石子移至其父亲结点。问这个游戏先手是否必胜？
2、修改一个结点的石子数。
3、新建一个结点石子数为x，其父亲设为y（保证y已经建立）
n<=10^5

分析

首先我们要知道这个博弈游戏的结论。
有一种博弈叫做阶梯博弈，就是有编号为0到n的台阶，每个台阶上都有若干石头。每次可以选择一个台阶，然后拿至少一个石头放到编号为它-1的台阶上。不能走的算数。问先手必胜还是后手必胜。
有一个结论就是我们只要管编号为奇数的阶梯即可。因为如果对手从偶阶梯选了一堆石头，那么你也可以选择这一堆石头，那么这堆石头仍在偶阶梯上。如果选了了奇阶梯上的石头，就可以理解为这堆石头消失了。那么这就变为了一个十分经典的Nim游戏。
拓展到树上，由于只能选[1,p]个石头，考虑sg[x]=mex(sg[x-i]){1<=i<=min(x,m)}
通过找规律我们可以发现sg[i]=x%(p+1)
那么我们可以把所有石头%(p+1)，若先手必胜则当且仅当深度为奇数的节点的石头异或和不为0.

那现在就好办了。
由于是子树，还带插入，我们考虑用splay来维护dfs序。但由于dfs序是动态的，所以我们需要资瓷查找每个节点后面第一个深度不大于它的点，那么每个节点只要维护该子树内所有节点的异或和，深度为奇数的节点的异或和还有深度最小的点即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=100005;
const int inf=0x7fffffff;

int n,m,q,cnt,last[N],dep[N],dfn[N],sz,root,a[N];
struct tree{int odd,all,l,r,mnd,fa;}t[N];
struct edge{int to,next;}e[N*2];

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void dfs(int x,int fa)
{
    dep[x]=dep[fa]+1;dfn[++sz]=x;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa) continue;
        dfs(e[i].to,x);
    }
}

void updata(int d)
{
    t[d].mnd=min(t[t[d].l].mnd,min(t[t[d].r].mnd,dep[d]));
    t[d].odd=t[d].all=0;
    t[d].all=t[t[d].l].all^t[t[d].r].all^a[d];
    if ((dep[t[d].l]-dep[d])%2!=0) t[d].odd^=t[t[d].l].all^t[t[d].l].odd;
    else t[d].odd^=t[t[d].l].odd;
    if ((dep[t[d].r]-dep[d])%2!=0) t[d].odd^=t[t[d].r].all^t[t[d].r].odd;
    else t[d].odd^=t[t[d].r].odd;
}

void rttr(int x)
{
    int y=t[x].l;
    if (x==root) root=y;
    t[x].l=t[y].r;t[t[y].r].fa=x;
    if (x==t[t[x].fa].l) t[t[x].fa].l=y;
    else if (x==t[t[x].fa].r) t[t[x].fa].r=y;
    t[y].fa=t[x].fa;
    t[y].r=x;t[x].fa=y;
    updata(x);updata(y);
}

void rttl(int x)
{
    int y=t[x].r;
    if (x==root) root=y;
    t[x].r=t[y].l;t[t[y].l].fa=x;
    if (x==t[t[x].fa].l) t[t[x].fa].l=y;
    else if (x==t[t[x].fa].r) t[t[x].fa].r=y;
    t[y].fa=t[x].fa;
    t[y].l=x;t[x].fa=y;
    updata(x);updata(y);
}

void splay(int x,int y)
{
    while (t[x].fa!=y)
    {
        int p=t[x].fa,g=t[p].fa;
        if (g==y)
        {
            if (x==t[p].l) rttr(p);
            else rttl(p);
            break;
        }
        if (x==t[p].l)
            if (p==t[g].l) rttr(g),rttr(p);
            else rttr(p),rttl(g);
        else
            if (p==t[g].r) rttl(g),rttl(p);
            else rttl(p),rttr(g);
    }
}

int find(int x)
{
    int x1=x;
    x=t[x].r;
    while (x)
    {
        if (t[t[x].l].mnd<=dep[x1]) x=t[x].l;
        else if (dep[x]<=dep[x1]) return x;
        else x=t[x].r;
    }
}

int build(int l,int r)
{
    if (l==r)
    {
        updata(dfn[l]);
        return dfn[l];
    }
    int mid=(l+r)/2,x=dfn[mid];
    if (l<mid) t[x].l=build(l,mid-1),t[t[x].l].fa=x;
    if (r>mid) t[x].r=build(mid+1,r),t[t[x].r].fa=x;
    updata(x);
    return x;
}

int findls(int x)
{
    x=t[x].l;int y;
    while (x) y=x,x=t[x].r;
    return y;
}

int main()
{
    freopen("a.in","r",stdin);freopen("a.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]),a[i]%=(m+1);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);
    }
    dfn[++sz]=N-3;
    dfs(1,0);
    dfn[++sz]=N-2;t[0].mnd=inf;
    root=build(1,sz);
    scanf("%d",&q);
    int lastans=0;
    while (q--)
    {
        int op;
        scanf("%d",&op);
        if (op==1)
        {
            int x;
            scanf("%d",&x);
            //x^=lastans;
            splay(x,0);
            int y=find(x),x1=x;
            x=findls(x);
            splay(x,0);splay(y,x);splay(x1,y);
            if (t[t[y].l].odd) printf("MeiZ\n"),lastans++;
            else printf("GTY\n");
        }
        else if (op==2)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            //x^=lastans;y^=lastans;
            splay(x,0);
            a[x]=y%(m+1);
            updata(x);
        }
        else
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            //x^=lastans;y^=lastans;z^=lastans;
            dep[y]=dep[x]+1;a[y]=z%(m+1);root=y;
            splay(x,0);
            int r=t[x].r;
            t[x].r=0;t[x].fa=t[r].fa=y;
            t[y].l=x;t[y].r=r;
            updata(x);updata(y);
        }
    }
    return 0;
}