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;
}