题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3011
以1号点为根建树,每个点维护一个大根堆,堆里的元素的权值等于到当前点的距离,然后只要堆顶的值大于L,就弹顶。
当向父亲合并的时候需要给堆里的元素整个加上一个值,于是写个推标记的pushdown就可以了
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define LL long long
const int N=200005;
const int M=400005;
int n,ls[N],rs[N],cnt,to[M],nxt[M],lj[N],siz[N],ans[N];
LL dis[N],lazy[N],L,w[M];
int ro[N],d[N];
void ins(int f,int t,LL p) {cnt++,to[cnt]=t,nxt[cnt]=lj[f],lj[f]=cnt,w[cnt]=p;}
void add(int f,int t,LL p) {ins(f,t,p),ins(t,f,p);}
void pushup(int x)
{
siz[x]=siz[ls[x]]+siz[rs[x]]+1;
}
void Pdis(int x,LL v)
{
if(!x) return;
dis[x]+=v;
lazy[x]+=v;
}
void pushdown(int x)
{
if(lazy[x]) Pdis(ls[x],lazy[x]),Pdis(rs[x],lazy[x]),lazy[x]=0;
}
int merge(int x,int y)
{
if(!x) return y;
if(!y) return x;
if(dis[x]<dis[y]) swap(x,y);
pushdown(x);
rs[x]=merge(rs[x],y);
if(d[ls[x]]<d[rs[x]]) swap(ls[x],rs[x]);
d[x]=d[rs[x]]+1;
pushup(x);
return x;
}
void pop(int x)
{
pushdown(ro[x]);
ro[x]=merge(ls[ro[x]],rs[ro[x]]);
}
void dfs(int x,int fa)
{
ro[x]=x;
d[x]=siz[x]=1;
for(int i=lj[x];i;i=nxt[i])
if(to[i]!=fa)
{
dfs(to[i],x);
Pdis(ro[to[i]],w[i]);
while(dis[ro[to[i]]]>L) pop(to[i]);
ro[x]=merge(ro[x],ro[to[i]]);
}
ans[x]=siz[ro[x]];
}
int main()
{
scanf("%d%lld",&n,&L);
LL x,y;
for(int i=2;i<=n;i++)
{
scanf("%lld%lld",&x,&y);
add(x,i,y);
}
dfs(1,0);
for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
}