【SHTSC2012day2】魔法树(tree)

前言

：太久没有写blog了，早上做NOI网络赛，下午不用改题，于是就瞎扯。

题解

这题其实挺好做的，直接上树剖，然后我们有一个显然的结论，一个子树内部的tree值肯定是连续的，那么直接记录一下在线段树中左端点和右端点的位置即可。

#include<cstdio> 
#include<algorithm>
#define fo(i,a,b) for (int (i)=(a);(i)<=(b);(i)++)
#define Fo(i,x) for (int i=head[x];i;i=next[i])
const int N=100000+55;
typedef long long ll;
using namespace std;
int next[N],head[N],to[N],size[N],top[N],d[N],son[N],ls[N],rs[N],tree[N],fa[N];
char ch[8];
ll s[N*4],sum,a[N*4];
int y1,y2,w,cnt,n,tot;
void R(int &n)
{
    int t=0,p=1;char ch;
    for(ch=getchar();!('0'<=ch && ch<='9');ch=getchar())
        if(ch=='-') p=-1;
    for(;'0'<=ch && ch<='9';ch=getchar()) t=t*10+ch-'0';
    n=t*p;
}
void add(int x,int y)
{
    to[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}
void dfs(int x)
{
    size[x]=1;
    int v;
    Fo(i,x)
    {
        v=to[i];d[v]=d[x]+1;fa[v]=x;
        dfs(v);
        size[x]+=size[v];
        if (size[v]>size[son[x]])son[x]=v;
    }
}
void Dfs(int x,int T)
{
    ls[x]=rs[x]=tree[x]=++cnt;
    top[x]=T;
    int v;
    if (son[x]) Dfs(son[x],T),ls[x]=min(ls[x],ls[son[x]]),rs[x]=max(rs[x],rs[son[x]]);
    else return;
    Fo(i,x)
    {
        v=to[i];
        if (v==son[x]) continue;
        Dfs(v,v);
        ls[x]=min(ls[x],ls[v]),rs[x]=max(rs[x],rs[v]);
    }
}
void update(int o,int l,int r)
{
    if (y1<=l&&r<=y2)
    {
        a[o]+=w;
    }
    else
    {
        int m=l+r>>1;
        if (y1<=m) update(o<<1,l,m);
        if (m<y2) update(o<<1|1,m+1,r);
    }
    s[o]=0;
    if (r>l) s[o]=s[o<<1]+s[o<<1|1];
    if (a[o]) s[o]+=a[o]*(r-l+1);
}
void query(int o,int l,int r,ll add)
{
    if (y1<=l&&r<=y2)
    {
        sum+=1ll*add*(r-l+1)+s[o];
        return;
    }
    int m=l+r>>1;
    if (y1<=m)query(o<<1,l,m,add+a[o]);
    if (m<y2)query(o<<1|1,m+1,r,add+a[o]);
}
void get(int x,int y)
{
    while (top[x]!=top[y])
    {
        if (d[top[x]]<d[top[y]]) swap(x,y);
        y1=tree[top[x]];y2=tree[x];
        update(1,1,n);
        x=fa[top[x]];
    }
    if (d[x]<d[y]) swap(x,y);
    y1=tree[y];y2=tree[x];
    update(1,1,n);
}
int main()
{
    //freopen("jzoj2893.in","r",stdin);
    //freopen("jzoj2893.out","w",stdout);
    scanf("%d",&n);
    int x,y,q;
    fo(i,1,n-1)
    {
        scanf("%d%d",&x,&y);
        add(x+1,y+1);
    }
    dfs(1);
    Dfs(1,1);
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s",ch);
        if (ch[0]=='A') 
        {
            scanf("%d%d%d",&x,&y,&w);
            x++,y++;
            get(x,y);
        }
        else
        {
            scanf("%d",&x);x++;
            sum=0;
            y1=ls[x];y2=rs[x];
            query(1,1,n,0);
            printf("%lld\n",sum);
        }
    }
    return 0;
}