bzoj3730震波（點分樹，樹狀數組）

題目大意：n個節點的樹，m次操作，每次修改每個點權值，或詢問與節點x距離不超過k的所有點權值之和。

      對點分樹的每個節點X用個線段樹維或樹狀數組維護其子樹中每個點到X的距離權值，再用另一個維護到fa[x]的距離權值（用於查詢時容斥）。修改詢問時沿着父節點往上跑 log n 層更新或查詢。查詢時需要減去前一層中的結果。。

    一開始用動態開節點的線段樹+LCA玄學CE的了好幾發（到現在也還不知道CE原因。。orz），換了個寫法後不是T就是RE（心態逐步炸裂），然後不知羞恥的上網看了一下午別人的做法，將線段樹換爲樹狀數組後又RE了幾發（心態完全炸裂），調到晚上最後終於過了。。菜枯了。

    代碼：

    點分樹+樹狀數組 。

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<vector>
using namespace std;
int A[100010],B[100010],vis[100010],ma[100010],siz[100010],root;
fa[100010][30],dep[100010],dis[100010][30],n;
vector<int>g[100010];
vector<int>T[2][100010];
void dfsroot(int u,int f,int sum){
     int i,v;
     siz[u]=1;
     ma[u]=0;
     for(i=0;i<g[u].size();i++){
         v=g[u][i];
         if(v!=f&&vis[v]==0){
              dfsroot(v,u,sum);
              siz[u]+=siz[v];
              if(ma[u]<siz[v]) ma[u]=siz[v];
         }
     }
     ma[u]=max(sum-siz[u],ma[u]);
     if(root==0||ma[u]<ma[root]) root=u;
}
void dfsdis(int u,int f,int k,int d){
     int i,v;
     dep[u]++;
     fa[u][dep[u]]=k;
     dis[u][dep[u]]=d;
     siz[u]=1;
     for(i=0;i<g[u].size();i++){
         v=g[u][i];
         if(v!=f&&vis[v]==0){
             dfsdis(v,u,k,d+1);
             siz[u]+=siz[v];
         }
     }
}
void dfs(int u){
     int i,v;
     vis[u]=1;
     dfsdis(u,0,u,0);
     T[0][u].resize(siz[u]+5);
     T[1][u].resize(siz[u]+5);
     for(i=0;i<g[u].size();i++){
         v=g[u][i];
         if(vis[v]==0){
              root=0;
              dfsroot(v,0,siz[v]);
              dfs(root);
         }
     }
}

int lb(int x){
    return x&-x;
}
void add(int u,int v,int k,int id){
     for(;u<T[id][k].size();u+=lb(u)) T[id][k][u]+=v;
}
int qsum(int u,int k,int id){
     int sum=0;
     for(;u;u-=lb(u)) sum+=T[id][k][u];
     return sum;
}

void update(int u,int k){
    int i,f,d;
    for(i=dep[u];i;i--){
        f=fa[u][i];
        add(dis[u][i]+1,k-A[u],f,0);
        if(i>1) add(dis[u][i-1]+1,k-A[u],f,1);
    }
}
int fans(int u,int k){
    int i,v,f,f1,d,dd,ans=0;
    for(i=dep[u];i;i--){
        f=fa[u][i];
        f1=fa[u][i+1];
        d=dis[u][i];
        if(d<=k){
            dd=T[0][f].size()-1;
            dd=min(dd,k-d+1);
            ans+=qsum(dd,f,0);
            if(f1){
                dd=T[1][f1].size()-1;
                dd=min(dd,k-d+1);
                ans-=qsum(dd,f1,1);
            }
        }
    }
    return ans;
}
int main(){
    int a,b,c,ans=0,i,m;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)  scanf("%d",&B[i]);
    for(i=1;i<n;i++){
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
    }
    root=0;
    dfsroot(1,0,n);
    dfs(root);
    for(i=1;i<=n;i++){
        update(i,B[i]);
        A[i]=B[i];
    }
    while(m--){
        scanf("%d%d%d",&a,&b,&c);
        b^=ans;
        c^=ans;
        if(a){
            update(b,c);
            A[b]=c;
        }
        else{
            ans=fans(b,c);
            printf("%d\n",ans);
        }
    }
    return 0;
}