bzoj 4372: 烁烁的游戏 （动态点分治+线段树+LCA）

题目描述

传送门

题目大意：给一颗n个节点的树，边权均为1，初始点权均为0，m次操作：
Q x：询问x的点权。
M x d w：将树上与节点x距离不超过d的节点的点权均加上w

题解

对于每个点开一棵线段树，线段树中维护到这个点距离为x的所有增量。
修改一个点的时候需要修改这个点到点分树的根路径上所有点的答案。
查询的时候也是一样。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define inf 1000000000
#define N 200003
using namespace std;
struct data{
  int ls,rs,x;
}tr[N*60];
int n,m,root[N],root1[N],point[N],nxt[N],v[N],ans,sum,rt,tot,sz;
int f[N],mi[20],fa[N][20],size[N],deep[N],vis[N],belong[N];
void getroot(int x,int fa)
{
    f[x]=0; size[x]=1;
    for (int i=point[x];i;i=nxt[i]) {
     if (v[i]==fa||vis[v[i]]) continue;
     getroot(v[i],x);
     size[x]+=size[v[i]];
     f[x]=max(f[x],size[v[i]]);
    }
    f[x]=max(f[x],sum-size[x]);
    if (f[x]<f[rt]) rt=x;
}
void dfs(int x,int father)
{
    deep[x]=deep[father]+1;
    for (int i=1;i<=17;i++){
      if (deep[x]-mi[i]<0) break;
      fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for (int i=point[x];i;i=nxt[i]) {
      if (v[i]==father) continue;
      fa[v[i]][0]=x;
      dfs(v[i],x);
    }
}
int lca(int x,int y)
{
    if (deep[x]<deep[y]) swap(x,y);
    int k=deep[x]-deep[y];
    for (int i=0;i<=17;i++) 
     if ((k>>i)&1) x=fa[x][i];
    if (x==y) return x;
    for (int i=17;i>=0;i--)
     if (fa[x][i]!=fa[y][i]) 
      x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int dis(int x,int y)
{
    return deep[x]+deep[y]-2*deep[lca(x,y)];
}
void build(int x)
{
    vis[x]=1;
    for (int i=point[x];i;i=nxt[i]) {
       if (vis[v[i]]) continue;
       rt=0; sum=size[v[i]];
       getroot(v[i],x);
       belong[rt]=x;
       build(rt);
    }
}
void insert(int &i,int l,int r,int x,int d)
{
     if (!i) i=++sz;
     tr[i].x+=d;
     if (l==r) return;
     int mid=(l+r)/2;
     if (x<=mid) insert(tr[i].ls,l,mid,x,d); 
     else insert(tr[i].rs,mid+1,r,x,d);
}
int query(int now,int l,int r,int ll,int rr)
{
    if (ll<=l&&r<=rr) return tr[now].x;
    int mid=(l+r)/2;
    int ans=0;
    if (ll<=mid) ans+=query(tr[now].ls,l,mid,ll,rr);
    if (rr>mid) ans+=query(tr[now].rs,mid+1,r,ll,rr);
    return ans;
}
void change(int u,int x,int d,int w)
{
   int D=dis(u,x);
   if (d-D>=0) insert(root[u],0,n,d-D,w);
   //cout<<u<<" "<<d<<" "<<w<<endl;
   int v=belong[u];
   if (!v) return;
   D=dis(x,v);
   if (d-D>=0) insert(root1[u],0,n,d-D,w);
   change(v,x,d,w);
}
void calc(int u,int x)
{
   if (u==x) ans+=query(root[u],0,n,0,n);
   else {
      int d=dis(u,x); //cout<<u<<" "<<x<<" "<<d<<endl; 
      ans+=query(root[u],0,n,d,n);
      //cout<<u<<" "<<query(root[u],0,n,d,n)<<endl;
   }
   if (!belong[u]) return;
   int d=dis(belong[u],x);
   ans-=query(root1[u],0,n,d,n);
   calc(belong[u],x);
}
void add(int x,int y)
{
    tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    tot++; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
int main()
{
    freopen("a.in","r",stdin);
    freopen("my.out","w",stdout);
    mi[0]=1;
    for (int i=1;i<=18;i++) mi[i]=mi[i-1]*2;
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++) {
       int x,y; scanf("%d%d",&x,&y);
       add(x,y);
    }
    dfs(1,0);
    rt=0; f[0]=inf; sum=n;
    getroot(1,0); build(rt);
   // cout<<rt<<endl;
   // for (int i=1;i<=n;i++) cout<<belong[i]<<" "; cout<<endl;
    for (int i=1;i<=m;i++) {
      char s[10]; 
      int x,d,w;
      scanf("%s%d",s,&x);
      if (s[0]=='M') {
        scanf("%d%d",&d,&w);
        d=min(d,n);
        change(x,x,d,w);
       }
      else {
         ans=0;
         calc(x,x);
         printf("%d\n",ans);
      }
    }
}