bzoj 2836 魔法树——树链剖分

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=2836

树剖裸题。然而WA、RE了好久……

原来是跳 top 的那个地方！ top 不相等的时候比较的是 top 的深度，不是自己的深度！ top 相等之后才比较自己的深度。

dfs 序离开自己时的标号就是 dfn[ cr ] + siz[ cr ] -1 。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+;
int n,q,hd[N],xnt,to[N],nxt[N];
int fa[N],dep[N],dfn[N],tim,siz[N],son[N],top[N];
int tot,ls[N<<],rs[N<<];
ll sm[N<<],laz[N<<];//
char ch[];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return fx?ret:-ret;
}
void add(int x,int y)
{
  to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;
}
void dfs(int cr)
{
  dep[cr]=dep[fa[cr]]+; siz[cr]=;
  for(int i=hd[cr],v;i;i=nxt[i])
    {
      v=to[i];
      dfs(v);siz[cr]+=siz[v];
      siz[v]>siz[son[cr]]?son[cr]=v:;
    }
}
void dfsx(int cr)
{
  dfn[cr]=++tim;
  if(son[cr])top[son[cr]]=top[cr],dfsx(son[cr]);
  for(int i=hd[cr],v;i;i=nxt[i])
    if((v=to[i])!=son[cr])
      top[v]=v,dfsx(v);
}
void build(int l,int r,int cr)
{
  if(l==r)return; int mid=l+r>>;
  ls[cr]=++tot;build(l,mid,ls[cr]);
  rs[cr]=++tot;build(mid+,r,rs[cr]);
}
void pshd(int cr,int l,int mid,int r)
{
  if(!laz[cr])return; ll d=laz[cr]; laz[cr]=;
  sm[ls[cr]]+=d*(mid-l+);sm[rs[cr]]+=d*(r-mid);
  laz[ls[cr]]+=d;laz[rs[cr]]+=d;
}
void pshp(int cr){sm[cr]=sm[ls[cr]]+sm[rs[cr]];}
void mdfy(int l,int r,int cr,int L,int R,int k)
{
  if(l>=L&&r<=R){sm[cr]+=(ll)k*(r-l+);laz[cr]+=k;return;}
  int mid=l+r>>;pshd(cr,l,mid,r);
  if(L<=mid)mdfy(l,mid,ls[cr],L,R,k);
  if(mid<R)mdfy(mid+,r,rs[cr],L,R,k);
  pshp(cr);
}
void mdfy(int x,int y,int k)
{
  while(top[x]!=top[y])
    {
      if(dep[top[x]]<dep[top[y]])swap(x,y);//dep[top]!!!
      mdfy(,n,,dfn[top[x]],dfn[x],k);
      x=fa[top[x]];
    }
  if(dep[x]<dep[y])swap(x,y);//
  mdfy(,n,,dfn[y],dfn[x],k);
}
ll query(int l,int r,int cr,int L,int R)
{
  if(l>=L&&r<=R)return sm[cr];
  int mid=l+r>>;pshd(cr,l,mid,r);
  if(R<=mid)return query(l,mid,ls[cr],L,R);
  if(mid<L)return query(mid+,r,rs[cr],L,R);
  return query(l,mid,ls[cr],L,R)+query(mid+,r,rs[cr],L,R);
}
int main()
{
  n=rdn();
  for(int i=,u,v;i<n;i++)
    {
      u=rdn()+; v=rdn()+; fa[v]=u; add(u,v);
    }
  dfs(); top[]=; dfsx(); tot=; build(,n,);
  int u,v,d;q=rdn();
  while(q--)
    {
      scanf("%s",ch);
      if(ch[]=='A')
    {
      u=rdn()+; v=rdn()+; d=rdn(); mdfy(u,v,d);
    }
      else
    {
      u=rdn()+;
      printf("%lld\n",query(,n,,dfn[u],dfn[u]+siz[u]-));//
    }
    }
  return ;
}