【GDOI2016模拟3.15】染色

Description

给出一棵树，初始时所有点都是白色。每次操作把一个点染黑，或是询问所有黑点到这个点的距离和。
n,m<=10^5

Solution

首先，答案是$\sum_{i是黑色}{dis[i]+dis[x]-dis[lca(i,x)]}$
那么我们知道了所有黑色点的距离和，和黑色点的个数，剩下的就
和[bzoj3626][LNOI2014]LCA一样~(≧▽≦)/~啦啦啦
然后就愉快滴搞掂~(≧▽≦)/~啦啦啦（这个表情是搜狗自己跳出来的(⊙o⊙)哦）
感觉自己最近懒癌发作

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a) for(int i=last[a];i;i=next[i])
#define N 100005
using namespace std;
typedef long long ll;
struct note{ll v,add,size;}tr[N*3];
int t[N],next[N],last[N],size[N],son[N],w[N],top[N],fa[N],h[N];
int n,m,tot,l,bz,x;
ll sum[N],a[N],v[N],cnt,num;
bool color[N];
void add(int x,int y,ll z) {
    t[++l]=y;v[l]=z;next[l]=last[x];last[x]=l;
}
void dfs(int x) {
    size[x]=1;int k=0;
    rep(i,x) {
        sum[t[i]]=sum[x]+v[i];dfs(t[i]);size[x]+=size[t[i]];
        if (size[t[i]]>k) k=size[t[i]],son[x]=t[i];
    }
}
void make(int x,int y) {
    w[x]=++tot;top[x]=y;h[tot]=x;
    if (!son[x]) return;
    make(son[x],y);
    rep(i,x) if (t[i]!=son[x]) make(t[i],t[i]);
}
void back(int v,ll x) {
    tr[v].v+=tr[v].size*x;
    tr[v].add+=x;
}
void down(int v) {
    if (tr[v].add) {
        back(v*2,tr[v].add);
        back(v*2+1,tr[v].add);
        tr[v].add=0;
    }
}
void build(int v,int l,int r) {
    if (l==r) {tr[v].size=a[h[l]];return;}
    int m=(l+r)/2;
    build(v*2,l,m);build(v*2+1,m+1,r);
    tr[v].size=tr[v*2].size+tr[v*2+1].size;
}
void change(int v,int l,int r,int x,int y) {
    if (l==x&&r==y) {back(v,1);return;}
    int m=(l+r)/2;down(v);
    if (y<=m) change(v*2,l,m,x,y);
    else if (x>m) change(v*2+1,m+1,r,x,y);
    else change(v*2,l,m,x,m),change(v*2+1,m+1,r,m+1,y);
    tr[v].v=tr[v*2].v+tr[v*2+1].v;
} 
ll find(int v,int l,int r,int x,int y) {
    if (l==x&&r==y) return tr[v].v;
    int m=(l+r)/2;down(v);
    if (y<=m) return find(v*2,l,m,x,y);
    else if (x>m) return find(v*2+1,m+1,r,x,y);
    else return find(v*2,l,m,x,m)+find(v*2+1,m+1,r,m+1,y);
}
void revise(int x) {
    int f=top[x];
    while (x) {
        change(1,1,n,w[f],w[x]);
        x=fa[f];f=top[x];
    }
}
ll query(int x) {
    int f=top[x];ll ans=0;
    while (x) {
        ans+=find(1,1,n,w[f],w[x]);
        x=fa[f];f=top[x];
    }
    return ans;
}
int main() {
    scanf("%d%d",&n,&m);
    fo(i,2,n) scanf("%d",&fa[i]),fa[i]++;
    fo(i,2,n) scanf("%lld",&a[i]),add(fa[i],i,a[i]);
    dfs(1);make(1,1);build(1,1,n);
    fo(i,1,m) {
        scanf("%d%d",&bz,&x);x++;
        if (bz==1&&!color[x]) revise(x),num++,cnt+=sum[x],color[x]=1;
        else if (bz==2) printf("%lld\n",cnt+num*sum[x]-2*query(x));
    }
}

Ps:发现wyx的代码跑的又短又快

后来发现原来是个大暴力！
看起来出题人把数据随机化了，所以树的期望高度是log的！
怎么能这样~(>_<)~