BZOJ 2836: 魔法树

Posted 2020-09-16 北屿

Descrption

一棵树，给一条路径增加权值，询问子树和.\(n\leqslant 10^5\)

Solution

树链剖分.

划水...

Code

/**************************************************************
    Problem: 2836
    User: BeiYu
    Language: C++
    Result: Accepted
    Time:5424 ms
    Memory:19080 kb
****************************************************************/
 
#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
const int N = 100050;
 
inline int in(int x=0,char s=getchar()) { while(s>‘9‘||s<‘0‘)s=getchar();
    while(s>=‘0‘&&s<=‘9‘)x=x*10+s-‘0‘,s=getchar();return x; }
 
int n,q;
 
namespace Seg {
    #define mid ((l+r)>>1)
    #define lc (o<<1)
    #define rc (o<<1|1)
     
    ll d[N<<2],tg[N<<2];
     
    void T(int o,int l,int r,ll v) {
        d[o]+=(r-l+1)*v,tg[o]+=v;
    }
    void Push(int o,int l,int r) {
        if(l!=r) T(lc,l,mid,tg[o]),T(rc,mid+1,r,tg[o]);
        tg[o]=0;return;
    }
    void Update(int o) { d[o]=d[lc]+d[rc]; }
    void Add(int o,int l,int r,int L,int R,ll v) {
        Push(o,l,r);
        if(L<=l && r<=R) { T(o,l,r,v);return; }
        if(L<=mid) Add(lc,l,mid,L,R,v);
        if(R>mid) Add(rc,mid+1,r,L,R,v);
        Update(o);
    }
    ll Query(int o,int l,int r,int L,int R) {
        Push(o,l,r);
        if(L<=l && r<=R) return d[o];
        ll res=0;
        if(L<=mid) res+=Query(lc,l,mid,L,R);
        if(R>mid) res+=Query(rc,mid+1,r,L,R);
        return res;
    }
};
 
namespace Tree {
    vector<int> g[N];
    int cp,sz[N],tp[N],f[N],d[N],sn[N],pp[N],pd[N];
     
    void AddEdge(int u,int v) { g[u].push_back(v),g[v].push_back(u); }
    void DFS1(int u,int fa) {
        f[u]=fa,sz[u]=1,sn[u]=0,d[u]=d[fa]+1;
        for(int i=0,v;i<(int)g[u].size();i++) if((v=g[u][i])!=fa) {
            DFS1(v,u),sz[u]+=sz[v];
            if(!sn[u] || sz[sn[u]]<sz[v]) sn[u]=v;
        }
    }
    void DFS2(int u,int ttp) {
        pp[u]=++cp,tp[u]=ttp;
        if(sn[u]) DFS2(sn[u],ttp);
        for(int i=0,v;i<(int)g[u].size();i++)
            if((v=g[u][i])!=f[u] && v!=sn[u]) DFS2(v,v);
        pd[u]=cp;
    }
    ll Query(int x) { return Seg::Query(1,1,n,pp[x],pd[x]); }
    void Add(int u,int v,ll x) {
        for(int f1=tp[u],f2=tp[v];f1^f2;) {
            if(d[f1]<d[f2]) swap(f1,f2),swap(u,v);
            Seg::Add(1,1,n,pp[f1],pp[u],x);
            u=f[f1],f1=tp[u];
        }
        if(d[u]>d[v]) swap(u,v);
        Seg::Add(1,1,n,pp[u],pp[v],x);
    }
}
 
int main() {
    n=in();
    for(int i=1,u,v;i<n;i++) u=in(),v=in(),Tree::AddEdge(u,v);
    Tree::DFS1(0,0),Tree::DFS2(0,0);
    for(q=in();q--;) {
        char opt[10];
        int u,v,d;
        scanf("%s",opt);
        if(opt[0]==‘A‘) u=in(),v=in(),d=in(),Tree::Add(u,v,d);
        else d=in(),printf("%lld\n",Tree::Query(d));
    }
    return 0;
}