BZOJ 4372: 烁烁的游戏(动态点分治)

4372: 烁烁的游戏

Time Limit: 30 Sec  Memory Limit: 512 MB
Submit: 804  Solved: 288
[Submit][Status][Discuss]

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节点的皮皮鼠数量。

思路：
和震波那题差不多，按照分治结构建一棵树，
然后开两个动态线段树来维护。
代码：
#include<bits/stdc++.h>
using namespace std;
#define cherry main
const int maxn=1e5+10;
const int len=1e5;
int n,m;
struct node
{
    int lson,rson,sum;
}tree1[maxn*210],tree2[maxn*210];
vector<int>G[maxn];
int f[maxn][20],dist[maxn][20],dep[maxn],c1[maxn],c2[maxn];
int root,now_size,son[maxn],T1[maxn],T2[maxn],tol1,tol2;
bool vis[maxn];
void get_root(int v,int fa,int SIZE)
{
    int ma=0;
    son[v]=1;
    for(int i=0;i<G[v].size();i++)
    {
        int to=G[v][i];
        if(to==fa||vis[to]) continue;
        get_root(to,v,SIZE);
        son[v]+=son[to];
        ma=max(ma,son[to]);
    }
    ma=max(ma,SIZE-son[v]);
    if(ma<now_size)
    {
        now_size=ma;
        root=v;
    }
}
int creat_node1()
{
    tree1[++tol1].lson=tree1[tol1].rson=tree1[tol1].sum=0;
    return tol1;
}
int creat_node2()
{
    tree2[++tol2].lson=tree2[tol2].rson=tree2[tol2].sum=0;
    return tol2;
}
void get_dep(int v,int fa,int father,int d)
{
    for(int i=0;i<G[v].size();i++)
    {
        int to=G[v][i];
        if(to==fa||vis[to]) continue;
        f[to][++dep[to]]=father;
        dist[to][dep[to]]=d;
        get_dep(to,v,father,d+1);
    }
}
void build(int v,int fa)
{
    vis[v]=1;
    T1[v]=creat_node1();
    T2[v]=creat_node2();
    get_dep(v,fa,v,1);
    for(int i=0;i<G[v].size();i++)
    {
        int to=G[v][i];
        if(to==fa||vis[to]) continue;
        get_root(root=to,v,now_size=son[to]);
        build(root,v);
    }
}
void add2(int L,int C,int l,int r,int rt)
{
    if(l==r)
    {
        tree2[rt].sum+=C;
        return;
    }
    int mid=(l+r)/2;
    if(L<=mid)
    {
        if(tree2[rt].lson==0)
            tree2[rt].lson=creat_node2();
        add2(L,C,l,mid,tree2[rt].lson);
    }
    else
    {
        if(tree2[rt].rson==0)
            tree2[rt].rson=creat_node2();
        add2(L,C,mid+1,r,tree2[rt].rson);
    }
    tree2[rt].sum=tree2[tree2[rt].lson].sum+tree2[tree2[rt].rson].sum;
}
void add1(int L,int C,int l,int r,int rt)
{
    if(l==r)
    {
        tree1[rt].sum+=C;
        return;
    }
    int mid=(l+r)/2;
    if(L<=mid)
    {
        if(tree1[rt].lson==0)
            tree1[rt].lson=creat_node1();
        add1(L,C,l,mid,tree1[rt].lson);
    }
    else
    {
        if(tree1[rt].rson==0)
            tree1[rt].rson=creat_node1();
        add1(L,C,mid+1,r,tree1[rt].rson);
    }
    tree1[rt].sum=tree1[tree1[rt].lson].sum+tree1[tree1[rt].rson].sum;
}
void update(int x,int y,int z)
{
    if(y)add1(min(y,len),z,1,len,T1[x]);
    else c1[x]+=z;
    if(dep[x]&&y>dist[x][dep[x]])
        add2(min(len,y-dist[x][dep[x]]),z,1,len,T2[x]);
    else if(dep[x]&&y==dist[x][dep[x]])
        c2[x]+=z;
    for(int i=dep[x];i>0;i--)
    {
        if(y>dist[x][i])
            add1(min(len,y-dist[x][i]),z,1,len,T1[f[x][i]]);
        else if(y==dist[x][i])
            c1[f[x][i]]+=z;
        if(i>1&&y>dist[x][i-1])
            add2(min(len,y-dist[x][i-1]),z,1,len,T2[f[x][i]]);
        else if(i>1&&y==dist[x][i-1])
            c2[f[x][i]]+=z;
    }
}
int sum1(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R) return tree1[rt].sum;
    int ans=0,mid=(l+r)/2;
    if(L<=mid) ans+=sum1(L,R,l,mid,tree1[rt].lson);
    if(R>mid) ans+=sum1(L,R,mid+1,r,tree1[rt].rson);
    return ans;
}
int sum2(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R) return tree2[rt].sum;
    int ans=0,mid=(l+r)/2;
    if(L<=mid) ans+=sum2(L,R,l,mid,tree2[rt].lson);
    if(R>mid) ans+=sum2(L,R,mid+1,r,tree2[rt].rson);
    return ans;
}
int query(int x)
{
    int ans=sum1(1,len,1,len,T1[x])+c1[x];
    for(int i=dep[x];i>0;i--)
    {
        ans+=sum1(dist[x][i],len,1,len,T1[f[x][i]]);
        ans-=sum2(dist[x][i],len,1,len,T2[f[x][i+1]]);
    }
    return ans;
}
int cherry()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    get_root(root=1,0,now_size=n);
    build(root,0);
    for(int i=1;i<=n;i++) f[i][dep[i]+1]=i;
    while(m--)
    {
        char t[2];int x,y,z;
        scanf("%s",&t);
        if(t[0]=='Q')
        {
            scanf("%d",&x);
            printf("%d\n",query(x));
        }
        else
        {
            scanf("%d%d%d",&x,&y,&z);
            update(x,y,z);
        }
    }
    return 0;
}