BZOJ 4372 烁烁的游戏

Description

背景：烁烁很喜欢爬树，这吓坏了树上的皮皮鼠。
题意：
给定一颗n个节点的树，边权均为1，初始树上没有皮皮鼠。
烁烁他每次会跳到一个节点u，把周围与他距离不超过d的节点各吸引出w只皮皮鼠。皮皮鼠会被烁烁吸引，所以会一直待在节点上不动。
烁烁很好奇，在当前时刻，节点u有多少个他的好朋友---皮皮鼠。
大意：
给一颗n个节点的树，边权均为1，初始点权均为0，m次操作：
Q x：询问x的点权。
M x d w：将树上与节点x距离不超过d的节点的点权均加上w。

Input

第一行两个正整数：n，m
接下来的n-1行，每行三个正整数u，v，代表u，v之间有一条边。
接下来的m行，每行给出上述两种操作中的一种。

Output

对于每个Q操作，输出当前x节点的皮皮鼠数量。

Sample Input

7 6
1 2
1 4
1 5
2 3
2 7
5 6
M 1 1 2
Q 5
M 2 2 3
Q 3
M 1 2 1
Q 2

Sample Output

2
3
6

和震波哪题很像，也是线段树+动态点分治

只不过将操作换了，一样的原理

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 #define MAX 120000
 struct Node
 {
   int next,to;
 }edge[MAX<<];
 int num,head[MAX];
 int dep[MAX],fa[MAX],pos,size[MAX],minsize,root;
 int FA[MAX],son[MAX],top[MAX],Size;
 int lazy[MAX*];
 int ch[MAX*][],RT[MAX<<],n,m;
 int ans,val[MAX];
 bool vis[MAX];
 inline int gi()
 {
   int x=,flag=;
   char ch=getchar();
   while (ch<''||ch>'')
     {
       if (ch=='-') flag=-;
       ch=getchar();
     }
   while (ch>=''&&ch<='')
     {
       x=(x<<)+(x<<)+ch-'';
       ch=getchar();
     }
   return x*flag;
 }
 inline void add(int u,int v)
 {
   num++;
   edge[num].next=head[u];
   head[u]=num;
   edge[num].to=v;
 }
 void dfs1(int x,int pa)
 {
   size[x]=;
   dep[x]=dep[pa]+;
   for (int i=head[x];i;i=edge[i].next)
     {
       int v=edge[i].to;
       if (v==pa) continue;
       fa[v]=x;
       dfs1(v,x);
       size[x]+=size[v];
       if (size[son[x]]<size[v])
       son[x]=v;
     }
 }
 void dfs2(int x,int tp)
 {
   top[x]=tp;
   if (son[x])
     dfs2(son[x],tp);
   for (int i=head[x];i;i=edge[i].next)
     {
       int v=edge[i].to;
       if (v==fa[x]||v==son[x]) continue;
       dfs2(v,v);
     }
 }
 int lca(int x,int y)
 {
   while (top[x]!=top[y])
     {
       if (dep[top[x]]<dep[top[y]]) swap(x,y);
       x=fa[top[x]];
     }
   if (dep[x]<dep[y]) return x;
   return y;
 }
 int dis(int x,int y)
 {
   return dep[x]+dep[y]-(dep[lca(x,y)]<<);
 }
 void get_root(int x,int pa)
 {
   size[x]=;
   int ret=;
   for (int i=head[x];i;i=edge[i].next)
     {
       int v=edge[i].to;
       if (v==pa||vis[v]) continue;
       get_root(v,x);
       size[x]+=size[v];
       if (size[v]>ret) ret=size[v];
     }
   if (Size-size[x]>ret) ret=Size-size[x];
   if (ret<minsize) minsize=ret,root=x;
 }
 void solve(int x,int pa)
 {
   vis[x]=;
   FA[x]=pa;
   for (int i=head[x];i;i=edge[i].next)
     {
       int v=edge[i].to;
       if (vis[v]) continue;
       minsize=n;Size=size[v];
       get_root(v,x);
       solve(root,x);
     }
 }
 void update(int &rt,int l,int r,int L,int R,int w)
 {
   if (!rt) rt=++pos;
   if (l>=L&&r<=R)
   {lazy[rt]+=w;return;}
   int mid=(l+r)>>;
   if (L<=mid) update(ch[rt][],l,mid,L,R,w);
   if (R>mid) update(ch[rt][],mid+,r,L,R,w);
 }
 int query(int rt,int l,int r,int x)
 {
   if (!rt) return ;
   if (l==r)
     return lazy[rt];
   int mid=(l+r)>>;
   if (x<=mid) return query(ch[rt][],l,mid,x)+lazy[rt];
   else return query(ch[rt][],mid+,r,x)+lazy[rt];
 }
 void change(int u,int k,int w)
 {
   update(RT[u],,n,,k,w);
   for (int i=u;FA[i];i=FA[i])
     {
       int d=dis(u,FA[i]);
       if (d>k) continue;
       update(RT[FA[i]],,n,,k-d,w);
       update(RT[i+n],,n,,k-d,w);
     }
 }
 void ask(int u)
 {
   ans+=query(RT[u],,n,);
   for (int i=u;FA[i];i=FA[i])
     {
       int d=dis(u,FA[i]);
       ans+=query(RT[FA[i]],,n,d);
       ans-=query(RT[i+n],,n,d);
     }
 }
 int main()
 {int u,v;
 char s[];
   cin>>n>>m;
   for (int i=;i<=n-;i++)
     {
       u=gi();v=gi();
       add(u,v);add(v,u);
     }
   dfs1(,);
   dfs2(,);
   minsize=Size=n;
   get_root(,);
   solve(root,);
   ans=;
   for (int i=;i<=m;i++)
     {
       scanf("%s",s);
       if (s[]=='Q')
         {
             ans=;
             int x=gi();
             ask(x);
             printf("%d\n",ans);
         }
       else
         {
             int u=gi(),k=gi(),w=gi();
             change(u,k,w);
         }
     }
   return ;
 }