BZOJ4372: 烁烁的游戏

Description

背景：烁烁很喜欢爬树，这吓坏了树上的皮皮鼠。
题意：
给定一颗n个节点的树，边权均为1，初始树上没有皮皮鼠。
烁烁他每次会跳到一个节点u，把周围与他距离不超过d的节点各吸引出w只皮皮鼠。皮皮鼠会被烁烁吸引，所以会一直待在节点上不动。
烁烁很好奇，在当前时刻，节点u有多少个他的好朋友---皮皮鼠。
大意：
给一颗n个节点的树，边权均为1，初始点权均为0，m次操作：
Q x：询问x的点权。
M x d w：将树上与节点x距离不超过d的节点的点权均加上w。

Input

第一行两个正整数：n，m
接下来的n-1行，每行三个正整数u，v，代表u，v之间有一条边。
接下来的m行，每行给出上述两种操作中的一种。

Output

对于每个Q操作，输出当前x节点的皮皮鼠数量。

Sample Input

7 6
1 2
1 4
1 5
2 3
2 7
5 6
M 1 1 2
Q 5
M 2 2 3
Q 3
M 1 2 1
Q 2

Sample Output

2
3
6

HINT

 

数据范围：

n，m<=10^5,|w|<=10^4

注意：w不一定为正整数，因为烁烁可能把皮皮鼠吓傻了。

 

裸的动态树分治，和震波很像啊（时限宽松多了）。

套个支持区间增加的线段树就行了。

数组开小结果RE2次（TAT）。

前面10s的太强了。

#include<cstdio>
#include<cctype>
#include<queue>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
inline int read() {
    int x=0,f=1;char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=getchar()) x=x*10+c-'0';
    return x*f;
}
const int maxn=200010;
const int maxnode=20000010;
int n,m,first[maxn],next[maxn<<1],to[maxn<<1],e;
void AddEdge(int u,int v) {
    to[++e]=v;next[e]=first[u];first[u]=e;
    to[++e]=u;next[e]=first[v];first[v]=e;
}
int val[maxn],mn[maxn<<1][20],Log[maxn<<1],dep[maxn],pos[maxn],cnt;
void dfs(int x,int fa) {
    dep[x]=dep[fa]+1;pos[x]=++cnt;mn[cnt][0]=dep[x];
    ren if(to[i]!=fa) dfs(to[i],x),mn[++cnt][0]=dep[x];
}
void init() {
    Log[0]=-1;
    rep(i,1,cnt) Log[i]=Log[i>>1]+1;
    for(int j=1;(1<<j)<=cnt;j++)
        for(int i=1;i+(1<<j)-1<=cnt;i++)
            mn[i][j]=min(mn[i][j-1],mn[i+(1<<j-1)][j-1]);
}
int dist(int x,int y) {
    int k,ans=dep[x]+dep[y];
    x=pos[x];y=pos[y];if(x>y) swap(x,y);k=Log[y-x+1];
    return ans-2*min(mn[x][k],mn[y-(1<<k)+1][k]);
}
int vis[maxn],f[maxn],s[maxn],root,size;
void getroot(int x,int fa) {
    int maxs=0;s[x]=1;
    ren if(to[i]!=fa&&!vis[to[i]]) {
        getroot(to[i],x);s[x]+=s[to[i]];
        maxs=max(maxs,s[to[i]]);
    }
    f[x]=max(maxs,size-s[x]);
    if(f[root]>f[x]) root=x;
}
int fa[maxn];
void build(int x) {
    vis[x]=1;
    ren if(!vis[to[i]]) {
        size=f[0]=s[to[i]];getroot(to[i],root=0);
        fa[root]=x;build(root);
    }
}
int ls[maxnode],rs[maxnode],addv[maxnode],ToT;
int root1[maxn],root2[maxn];
void update(int& o,int l,int r,int ql,int qr,int v) {
    if(!o) o=++ToT;
    if(ql<=l&&r<=qr) addv[o]+=v;
    else {
        int mid=l+r>>1;
        if(ql<=mid) update(ls[o],l,mid,ql,qr,v);
        if(qr>mid) update(rs[o],mid+1,r,ql,qr,v);
    }
}
int query(int& o,int l,int r,int pos,int cur) {
    cur+=addv[o];
    if(!o||pos<0||l==r) return cur;
    int mid=l+r>>1;
    if(pos<=mid) return query(ls[o],l,mid,pos,cur);
    return query(rs[o],mid+1,r,pos,cur);
}
void update(int x,int w,int v) {
    update(root1[x],0,n,0,w,v);
    for(int i=x;fa[i];i=fa[i]) {
        int D=dist(x,fa[i]);
        if(w>=D) {
            update(root1[fa[i]],0,n,0,w-D,v);
            update(root2[i],0,n,0,w-D,v);
        }
    }
}
int query(int x) {
    int res=query(root1[x],0,n,0,0);
    for(int i=x;fa[i];i=fa[i]) {
        int D=dist(x,fa[i]);
        res+=query(root1[fa[i]],0,n,D,0)-query(root2[i],0,n,D,0);
    }
    return res;
}
int main() {
    n=read();m=read();
    rep(i,2,n) AddEdge(read(),read());
    dfs(1,0);init();
    f[0]=size=n;getroot(1,root=0);
    build(root);
    while(m--) {
        char cmd[2];
        scanf("%s",cmd);
        if(cmd[0]=='Q') printf("%d\n",query(read()));
        else {
            int x=read(),w=read(),v=read();
            update(x,w,v);
        }
    }
    return 0;
}