bzoj 3011: [Usaco2012 Dec]Running Away From the Barn

题意：

给出以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]);
}