题意:
给出以1号点为根的一棵有根树,问每个点的子树中与它距离小于等于l的点有多少个。
题解:
左偏数模板题。
维护下dep就可以了。
code:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#define LL long long
using namespace std;
struct node{
int y,next;
LL c;
}a[200010];int len=0,last[200010];
int n;
LL L;
struct trnode{
int lc,rc,d;
LL c;
trnode() {lc=rc=d=0;}
}tr[200010];
int ans[200010],root[200010];
LL dep[200010];
void ins(int x,int y,LL c)
{
a[++len].y=y;a[len].c=c;
a[len].next=last[x];last[x]=len;
}
int merge(int x,int y)
{
if(!x||!y) return x+y;
if(tr[x].c<tr[y].c) swap(x,y);
tr[x].rc=merge(tr[x].rc,y);
if(tr[tr[x].rc].d>tr[tr[x].lc].d) swap(tr[x].lc,tr[x].rc);
tr[x].d=tr[tr[x].rc].d+1;
return x;
}
void dfs(int x,int fa,LL c)
{
dep[x]=dep[fa]+c;root[x]=x;
tr[x].c=dep[x];tr[x].d=1;ans[x]=1;
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
dfs(y,x,a[i].c);ans[x]+=ans[y];
root[x]=merge(root[x],root[y]);
while(tr[root[x]].c>dep[x]+L) ans[x]--,root[x]=merge(tr[root[x]].lc,tr[root[x]].rc);
}
}
int main()
{
scanf("%d %lld",&n,&L);
for(int i=2;i<=n;i++)
{
int fa;LL c;scanf("%d %lld",&fa,&c);
ins(fa,i,c);
}
dfs(1,0,0);
for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
}