hdu_4918_Query on the subtree(树的分治+树状数组)

题目链接：hdu_4918_Query on the subtree

题意：

给出一颗n个点的树，每个点有一个权值，有两种操作，一种是将某个点的权值修改为v，另一种是查询距离点u不超过d的点的权值和。

题解：

这里可以去膜膜鸟神的博客。

简单来说就是对树的每个重心建立两个树状数组，然后对于每个点修改就在每个重心的BIT中去修改，查询也在每个重心的BIT中查询，然后容斥一下，就得出答案。

每种操作的复杂度为log²n,这题的细节比较多，具体看代码。

 #include<bits/stdc++.h>
 #define F(i,a,b) for(int i=a;i<=b;i++)
 #define pb push_back
 using namespace std;
 const int N=1e5+;

 int n,q,g[N],nxt[N*],v[N*],ed,w[N],vis[N],id[N];
 int pool[*N],C_ed,pool_ed,sz[N],mi,mx[N],ROOT;
 char op[];

 struct node
 {
     int rt,subrt,dis;
     node(){}
     node(int _rt,int _subrt,int _dis):rt(_rt),subrt(_subrt),dis(_dis){}
 }tmp;
 vector<node>vt[N];

 void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}
 void init(){ed=C_ed=pool_ed=;F(i,,n)vt[i].clear(),vis[i]=g[i]=;}
 inline void up(int &a,int b){if(a<b)a=b;}

 struct BIT
 {
     int *C,n;
     void init(int tot){n=tot,C=pool+pool_ed,pool_ed+=tot+;F(i,,n)C[i]=;}
     inline void add(int x,int c){while(x<=n)C[x]+=c,x+=x&-x;}
     inline int ask(int x,int an=)
     {
         if(x>n)x=n;
         while(x>)an+=C[x],x-=x&-x;
         return an;
     }
 }tr[N*];

 void get_rt(int u,int fa,int num)
 {
     sz[u]=,mx[u]=;
     for(int i=g[u];i;i=nxt[i])
         if(!vis[v[i]]&&v[i]!=fa)
         {
             get_rt(v[i],u,num);
             sz[u]+=sz[v[i]],up(mx[u],sz[v[i]]);
         }
     up(mx[u],num-sz[u]);
     if(mx[u]<mi)ROOT=u,mi=mx[u];
 }

 void del(int u,int fa,int rt,int subrt,int dis=)//将子树的每个点放进对应的重心
 {
     vt[u].pb(node(rt,subrt,dis));
     tr[rt].add(dis+,w[u]);
     tr[subrt].add(dis+,w[u]);
     for(int i=g[u];i;i=nxt[i])
         if(v[i]!=fa&&!vis[v[i]])
             del(v[i],u,rt,subrt,dis+);
 }

 void init_tree(int u=,int num=n)
 {
     mi=N,get_rt(u,u,num);
     int rt=ROOT,rt_id=++C_ed;
     tr[C_ed].init(sz[u]+);
     vis[rt]=,vt[rt].pb(node(C_ed,,));
     tr[C_ed].add(,w[rt]);
     get_rt(rt,rt,num);//从新计算sz的大小
     for(int i=g[rt];i;i=nxt[i])
         if(!vis[v[i]])
         {
             tr[++C_ed].init(sz[v[i]]+);
             del(v[i],v[i],rt_id,C_ed);
         }
     for(int i=g[rt];i;i=nxt[i])
         if(!vis[v[i]])
             init_tree(v[i],sz[v[i]]);
 }

 int main()
 {
     while(~scanf("%d%d",&n,&q))
     {
         init();
         F(i,,n)scanf("%d",w+i);
         F(i,,n-)
         {
             int x,y;
             scanf("%d%d",&x,&y);
             adg(x,y),adg(y,x);
         }
         init_tree();
         while(q--)
         {
             int u,v;
             scanf("%s%d%d",op,&u,&v);
             if(op[]=='!')
             {
                 int size=vt[u].size(),d=v-w[u];
                 F(i,,size-)
                 {
                     tmp=vt[u][i];
                     tr[tmp.rt].add(tmp.dis+,d);//dis+1去掉距离为0在BIT上超时的BUG
                     if(tmp.subrt)tr[tmp.subrt].add(tmp.dis+,d);
                 }
                 w[u]+=d;
             }else
             {
                 int d=v,ans=,size=vt[u].size();
                 F(i,,size-)
                 {
                     tmp=vt[u][i];
                     ans+=tr[tmp.rt].ask(d-tmp.dis+);
                     if(tmp.subrt)ans-=tr[tmp.subrt].ask(d-tmp.dis+);
                 }
                 printf("%d\n",ans);
             }
         }
     }
     return ;
 }