bzoj4372 烁烁的游戏（点分治+线段树）

无限CE…一丝丝怀疑人生qaq

考虑点分树维护这个东西。每个节点x开一颗线段树，每个叶子节点i记录距离x距离为i的点加了多少。每次修改/询问在点分树上跳父亲就好了。然后还要容斥一下多加的部分，因此需要每个点再开一颗线段树表示到fa[x]距离为i的要减掉多少。
可以吧前缀加单点询问变成单点加后缀询问，这样节点可以少一些。
复杂度$O(nlog^2n)$

空间应该也是$O(n2log^2n)$的，开不下，其实应该可以卡掉，至少我卡到了$log^2n$

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(T==S){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
inline bool get_op(){
    char ch=gc();
    while(ch!='M'&&ch!='Q') ch=gc();
    return ch=='M';
}
int n,m,h[N],num=0,Log[N<<1],mn[N<<1][20],dep[N],pos[N],rt,sumsz,sz[N],f[N],Gfa[N],dfn=0,rt1[N],rt2[N],owo=0;
struct edge{
    int to,next;
}data[N<<1];
struct node{
    int ls,rs,sum;
}tr[N*350];
bool vis[N];
inline void dfs(int x,int Fa){
    mn[++dfn][0]=x;pos[x]=dfn;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;
        dep[y]=dep[x]+1;dfs(y,x);mn[++dfn][0]=x;
    }
}
inline int Min(int x,int y){return dep[x]<dep[y]?x:y;}
inline void initrmq(){
    Log[0]=-1;
    for(int i=1;i<=dfn;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;i<=Log[dfn];++i)
        for(int j=1;j<=dfn;++j){
            if(j+(1<<(i-1))>dfn) break;
            mn[j][i]=Min(mn[j][i-1],mn[j+(1<<(i-1))][i-1]);
        }
}
inline void dfs1(int x,int Fa){
    sz[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]||y==Fa) continue; 
        dfs1(y,x);sz[x]+=sz[y];
    }
}
inline void dfs2(int x,int Fa){
    f[x]=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]||y==Fa) continue; 
        dfs2(y,x);f[x]=max(f[x],sz[y]);
    }f[x]=max(f[x],sumsz-sz[x]);if(f[x]<f[rt]) rt=x;
}
inline void solve(int x){
    vis[x]=1;dfs1(x,0);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;
        sumsz=sz[y];rt=0;dfs2(y,x);Gfa[rt]=x;solve(rt);
    }
}
inline int lca(int x,int y){
    x=pos[x];y=pos[y];if(x>y) swap(x,y);
    int d=Log[y-x+1];return Min(mn[x][d],mn[y-(1<<d)+1][d]);
}
inline int dis(int x,int y){return dep[x]+dep[y]-2*dep[lca(x,y)];}
inline void add(int &p,int l,int r,int x,int val){
    if(!p) p=++owo;tr[p].sum+=val;if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) add(tr[p].ls,l,mid,x,val);
    else add(tr[p].rs,mid+1,r,x,val);
}
inline int qsum(int p,int l,int r,int x,int y){
    if(!p) return 0;if(x<=l&&r<=y) return tr[p].sum;
    int mid=(l+r)>>1,res=0;
    if(x<=mid) res+=qsum(tr[p].ls,l,mid,x,y);
    if(y>mid) res+=qsum(tr[p].rs,mid+1,r,x,y);return res;
}
inline void modify(int u,int d,int val){
    for(int x=u;x;x=Gfa[x]){
        int dd=min(n,d-dis(u,x));if(dd>0) add(rt1[x],0,n,dd,val);
        if(!Gfa[x]) break;
        dd=min(n,d-dis(u,Gfa[x]));if(dd>0) add(rt2[x],0,n,dd,val);
    }
}
inline int ask(int u){
    int res=0;
    for(int x=u;x;x=Gfa[x]){
        int dd=dis(u,x);res+=qsum(rt1[x],0,n,dd,n);
        if(!Gfa[x]) break;
        dd=dis(u,Gfa[x]);res-=qsum(rt2[x],0,n,dd,n);
    }return res;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();f[0]=inf;
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1,0);initrmq();rt=0;sumsz=n;dfs1(1,0);dfs2(1,0);solve(rt);
    while(m--){
        if(get_op()){
            int x=read(),d=read(),val=read();
            modify(x,d,val);
        }else printf("%d\n",ask(read()));
    }return 0;
}