suoi08 一收一行破 (tarjanLca+树状数组)

用一个差分树状数组维护一下每个深度的和，然后每次拿着路径端点和lca加一加减一减就行了

#include<bits/stdc++.h>
#define pa pair<int,int>
#define ll long long
using namespace std;
const int maxn=200020;

inline ll rd(){
    ll x=0;char c=getchar();int neg=1;
    while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();}
    while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*neg;
}

int N,M,L=1;
int eg[maxn*2][2],egh[maxn],ect;
int dep[maxn],fa[maxn],v[maxn];
int qu[maxn*2][2],qh[maxn],op[maxn][3];
ll tr[maxn];
bool flag[maxn];

inline int lowbit(int x){return x&(-x);}
inline int getfa(int x){return x==fa[x]?x:fa[x]=getfa(fa[x]);}
inline void adeg(int a,int b){
    eg[++ect][0]=b;eg[ect][1]=egh[a];egh[a]=ect;
}inline void adq(int a,int b,int i){
    qu[i][0]=b;qu[i][1]=qh[a];qh[a]=i;
}

inline void add(int x,int y){
    for(;x&&x<=L;x+=lowbit(x)) tr[x]+=y;
}
inline ll query(int x){
    ll re=0;for(;x;x-=lowbit(x)) re+=tr[x];return re;
}

void tarjan(int x){
    flag[x]=1;
    for(int i=egh[x];i!=-1;i=eg[i][1]){
        if(flag[eg[i][0]]) continue;
        dep[eg[i][0]]=dep[x]+1;L=max(L,dep[x]+1);
        tarjan(eg[i][0]);fa[getfa(eg[i][0])]=getfa(x);
    }
    for(int i=qh[x];i!=-1;i=qu[i][1]){
        if(flag[qu[i][0]]) op[i>>1][0]=getfa(qu[i][0]);
    }
}



int main(){
    int i,j,k;
    N=rd(),M=rd();
    memset(egh,-1,sizeof(egh));memset(qh,-1,sizeof(qh));
    for(i=1;i<=N;i++) v[i]=rd();
    for(i=1;i<N;i++){
        int a=rd(),b=rd();
        adeg(a,b);adeg(b,a);
    }for(i=1;i<=M;i++){
        int a=rd(),b=rd();
        if(a==1){
            int c=rd();adq(b,c,i<<1);adq(c,b,i<<1|1);
            op[i][1]=b,op[i][2]=c;
        }else op[i][0]=b;
    }for(i=1;i<=N;i++) fa[i]=i;
    dep[1]=1;tarjan(1);//printf("ll");
    for(i=1;i<=N;i++){
        add(dep[i],v[i]);add(dep[i]+1,-v[i]);
    }
    for(i=1;i<=M;i++){
        if(op[i][1]){
            add(dep[op[i][0]],1);add(dep[op[i][0]]+1,-1);
            if(op[i][0]!=op[i][1]) add(dep[op[i][0]]+1,1),add(dep[op[i][1]]+1,-1);
            if(op[i][0]!=op[i][2]) add(dep[op[i][0]]+1,1),add(dep[op[i][2]]+1,-1);
        }else{
            if(op[i][0]>L) printf("0
");
            else printf("%lld
",query(op[i][0]));
        }
    }
    
    return 0;
}