【SHTSC2014】三叉神经树

【SHTSC2014】三叉神经树(neuron)

(File IO): input:neuron.in output:neuron.out
Time Limits: 8000 ms Memory Limits: 262144 KB
Description

Input

Output
输出 q行每一个整数，对应第 i次外界输入变化后的根细胞的输出。

Sample Input

3
2 3 4
5 6 7
8 9 10
0 0 0 0 1 1 1
5
4
4
5
6
8

Sample Output

1
0
0
1
1

Data Constraint

---

题意：

给一棵根为1的三叉(傻叉神经树)树，非叶子节点必有三个儿子，每个节点会储存01信息，每次只改变一个叶子节点的信息，某非叶子节点的01值会相应改变，取值为三个儿子的信息中较多的那个
比如，某节点x三个儿子信息为011，则x的信息为1
问每次改变后根的信息。

---

Solution

同一场比赛中，好像只会做这题。。。
8000ms好大，n=500000，$log^2$??
想想先。

记一个节点x的信息为1的儿子个数，t[x]
当t[x]>1时x的信息为1反之为0。
考虑某个t[x]的改变

1. 0->1，由于原本x不对fa[x]产生贡献，改变后t[fa[x]]不会加1，改变不会上传
2. 1->2，原本对fa[x]无贡献，改变后因为t[x]>1，t[fa[x]]++，改变上传
3. 2->3，原本对fa[x]已有贡献，改变后t[x]依然大于1，t[fa[x]]不会加1，改变不上传
4. 3->2，由于原本x对fa[x]产生贡献，改变后t[x]依然大于1，t[fa[x]]不会减1，改变不会上传
5. 2->1，原本对fa[x]有贡献，改变后因为t[x]<2，t[fa[x]]–，改变上传
6. 1->0，原本对fa[x]已没有贡献，改变后t[x]依然小于2，t[fa[x]]不会减1，改变不上传

经过这样分析可得：
每次某个叶子的改变，只会影响到它father开始，连续一段为1(或2)的点及其顶端的father

用树链剖（刨）分或LCT（个人推荐，更快）即可。

#include<cstring>
#include<cstdio>
#define N 1500100

using namespace std;

int fa[N],pre[N],son[N][2],val[N],add[N],up[4][N],sta[N],n,q,siz[N],stack[N],T;

int pd(int x){return son[fa[x]][1]==x;}
void update(int w){
    if(!w)return;
    int l=son[w][0],r=son[w][1];siz[w]=siz[l]+siz[r]+1;
    for(int k=0;k<4;k++){
        up[k][w]=up[k][r];
        if(up[k][r]==siz[r] && val[w]==k)up[k][w]=up[k][l]+1+siz[r];
    }
}
void rotate(int x){
    int y=fa[x],w=pd(x);
    fa[x]=fa[y];son[fa[x]][pd(y)]=x;fa[y]=x;pre[x]=pre[y];pre[y]=0;
    son[y][w]=son[x][1^w];fa[son[y][w]]=y;son[x][1^w]=y;update(y);
}
void inc(int x,int k){
    if(!x)return;
    val[x]+=k;add[x]+=k;
    if(k>0)for(int j=3;j>=k;--j)up[j][x]=up[j-k][x],up[j-k][x]=0;
    if(k<0)for(int j=0;j<4+k;j++)up[j][x]=up[j-k][x],up[j-k][x]=0;
}
void down(int x){
    if(!x)return;
    if(add[x])inc(son[x][0],add[x]),inc(son[x][1],add[x]),add[x]=0;
}
void rel(int x,int e){
    for(;x!=e;x=fa[x])stack[++T]=x;while(T)down(stack[T--]);
}
void splay(int x,int e){
    rel(x,e);
    for(;fa[x]!=e;rotate(x))if(fa[fa[x]]!=e)(pd(x)^pd(fa[x]))?rotate(x):rotate(fa[x]);
    update(x);
}
void access(int x){
    for(int y=0;x;y=x,x=pre[x]){
        splay(x,0);
        fa[son[x][1]]=0;pre[son[x][1]]=x;
        son[x][1]=y;pre[y]=0;fa[y]=x;update(x);
    }
}
void link(int x,int y){
    pre[x]=y;
}
int kth(int x,int k){
    for(int i=x;;){
        if(siz[son[i][0]]>=k)i=son[i][0];else
        if(siz[son[i][0]]+1==k)return i;else
        k-=siz[son[i][0]]+1,i=son[i][1];
    }
}
void turn1(int x){
    access(x);splay(x,0);int t=siz[x],k=up[1][x]+1;if(k>t)k=t;k=t-k;
    if(k==0)inc(x,1);else{
        int y=kth(x,k);
        splay(y,0);splay(x,y);inc(x,1);
    }
}
void turn0(int x){
    access(x);splay(x,0);int t=siz[x],k=up[2][x]+1;if(k>t)k=t;k=t-k;
    if(k==0)inc(x,-1);else{
        int y=kth(x,k);
        splay(y,0);splay(x,y);inc(x,-1);
    }
}

int main(){
    freopen("neuron.in","r",stdin);
    freopen("neuron.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)siz[i]=1,up[0][i]=1;
    for(int i=n+1;i<=3*n+1;i++)up[val[i]=1][i]=1,siz[i]=1;
    for(int i=1;i<=n;i++){
        int x1,x2,x3;scanf("%d %d %d",&x1,&x2,&x3);link(x1,i);link(x2,i);link(x3,i);
    }
    for(int i=n+1;i<=3*n+1;i++){
        int x;scanf("%d",&x);sta[i]=x;if(x)turn1(i);
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        int x;scanf("%d",&x);
        sta[x]^=1;(sta[x])?turn1(x):turn0(x);access(1);printf("%d\n",val[1]>=2);
    }
    fclose(stdin);fclose(stdout);
    return 0;
}