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;
}