[BZOJ4372]烁烁的游戏(动态点分治+线段树)

和[BZOJ3730]震波几乎一样，每个点建两棵线段树分别代表它的管辖范围内以它为LCA的路径的贡献和它对父亲的贡献。

注意点分树上的点的距离在原树上不单调，所以不能有若距离超出限制就break之类的判断。

 #include<cstdio>
 #include<algorithm>
 #define lson ls[x],L,mid
 #define rson rs[x],mid+1,R
 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 #define For(i,x) for (int i=h[x],k; i; i=nxt[i])
 using namespace std;

 const int N=,M=;
 char ch;
 int n,m,u,v,rt,x,d,w,nd,cnt,tim,dep[N],dfn[N],fa[N],lg2[N],sz[N];
 int h[N],to[N<<],nxt[N<<],sm[M],ls[M],rs[M],mn[N][],rt1[N],rt2[N],f[N],vis[N];

 void add(int u,int v){ to[++cnt]=v; nxt[cnt]=h[u]; h[u]=cnt; }

 void dfs(int x,int fa){
     dep[x]=dep[fa]+; mn[++tim][]=dep[x]; dfn[x]=tim;
     For(i,x) if ((k=to[i])!=fa) dfs(k,x),mn[++tim][]=dep[x];
 }

 int dis(int x,int y){
     int l=dfn[x],r=dfn[y];
     if (l>r) swap(l,r);
     int t=lg2[r-l+];
     return dep[x]+dep[y]-*min(mn[l][t],mn[r-(<<t)+][t]);
 }

 void find(int x,int F,int S,int &rt){
     sz[x]=; f[x]=;
     For(i,x) if ((k=to[i])!=F && !vis[k])
         find(k,x,S,rt),sz[x]+=sz[k],f[x]=max(f[x],sz[k]);
     f[x]=max(f[x],S-sz[x]);
     if (f[x]<f[rt]) rt=x;
 }

 void devide(int x){
     vis[x]=;
     For(i,x) if (!vis[k=to[i]]){
         int rt=; find(k,x,sz[k],rt); fa[rt]=x; devide(rt);
     }
 }

 void mdf(int &x,int L,int R,int pos,int k){
     if (!x) x=++nd; sm[x]+=k;
     if (L==R) return;
     int mid=(L+R)>>;
     if (pos<=mid) mdf(lson,pos,k); else mdf(rson,pos,k);
 }

 int que(int x,int L,int R,int pos){
     if (L==R) return sm[x];
     int mid=(L+R)>>;
     if (pos<=mid) return sm[rs[x]]+que(lson,pos);
         else return que(rson,pos);
 }

 void Mdf(int x,int d,int w){
     for (int i=x; i; i=fa[i]){
         if (d>=dis(i,x)) mdf(rt1[i],,n,d-dis(i,x),w);
         if (fa[i] && d>=dis(fa[i],x)) mdf(rt2[i],,n,d-dis(fa[i],x),w);
     }
 }

 int Que(int x){
     int res=;
     for (int i=x; i; i=fa[i]){
         res+=que(rt1[i],,n,dis(i,x));
         if (fa[i]) res-=que(rt2[i],,n,dis(fa[i],x));
     }
     return res;
 }

 int main(){
     freopen("bzoj4372.in","r",stdin);
     freopen("bzoj4372.out","w",stdout);
     scanf("%d%d",&n,&m);
     rep(i,,n) scanf("%d%d",&u,&v),add(u,v),add(v,u);
     dfs(,);
     rep(j,,) rep(i,,tim-(<<j)+)
         mn[i][j]=min(mn[i][j-],mn[i+(<<(j-))][j-]);
     lg2[]=; rep(i,,tim) lg2[i]=lg2[i>>]+;
     f[]=n+; find(,,n,rt); devide(rt);
     rep(i,,m){
         scanf(" %c",&ch);
         if (ch=='Q') scanf("%d",&x),printf("%d\n",Que(x));
             else scanf("%d%d%d",&x,&d,&w),Mdf(x,d,w);
     }
     return ;
 }