3011: Running Away From the Barn 可并堆/倍增+树链剖分+树上差分/dfs序+主席树

Description
给出以1号点为根的一棵有根树，问每个点的子树中与它距离小于**等于**l的点有多少个。

题解：

这道题本来是我在学可并堆的时候找到的，但是我并没有想出可并堆的做法，下面提供三种做法。

1、可并堆

首先预处理出每个点到根的距离$v[x]$，然后以此为关键字对每个节点先建个大根堆，然后dfs一次，每次把所有儿子和自己合并，然后一直删堆顶，直到满足条件为止，那么最后堆的size就是每个点的答案。

2、倍增+树链剖分+树上差分

预处理出每个点到第$2^i$个父亲的距离$v[x][i]$，然后对于每个点考虑它的贡献，往上找这个点最远能对哪个祖先做贡献，然后在这一段的$ans+1$，可以用树链剖分+树上差分实现，最后扫一遍就知道每个点的答案了。

3、dfs序+主席树

首先预处理出每个点到根的距离$v[x]$，然后在子树中找符合条件的有多少个就可以了。

代码：（可并堆）

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=200010;
const int inf=2147483647;
LL read()
{
    LL x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
struct Edge{int y,next;LL d;}e[Maxn];
int len=0,last[Maxn];
void ins(int x,int y,LL d){int t=++len;e[t].y=y;e[t].d=d;e[t].next=last[x];last[x]=t;}
int n,f[Maxn],size[Maxn],dis[Maxn],lc[Maxn],rc[Maxn];LL L,v[Maxn];
int Find(int x){return(f[x]=((f[x]==x)?x:Find(f[x])));}
void dfs(int x)
{
    f[x]=x;size[x]=1;dis[x]=0;
    for(int i=last[x];i;i=e[i].next)
    {
        int y=e[i].y;
        v[y]=v[x]+e[i].d;
        dfs(y);
    }
}
int merge(int A,int B)
{
    if(!A)return B;if(!B)return A;
    if(v[A]<v[B])swap(A,B);
    size[A]+=size[B];
    rc[A]=merge(rc[A],B);
    if(dis[rc[A]]>dis[lc[A]])swap(lc[A],rc[A]);
    dis[A]=dis[rc[A]]+1;
    return A;
}
int ans[Maxn];
void DFS(int x)
{
    for(int i=last[x];i;i=e[i].next)
    {
        int y=e[i].y;
        DFS(y);
        f[x]=Find(x),f[y]=Find(y);
        f[x]=f[y]=merge(f[x],f[y]);
    }
    while(1)
    {
        int t=Find(x);
        if(v[t]-v[x]>L)f[t]=f[lc[t]]=f[rc[t]]=merge(lc[t],rc[t]);
        else break;
    }
    ans[x]=size[Find(x)];
}
int main()
{
    dis[0]=-1;
    n=read(),L=read();
    for(int i=2;i<=n;i++)
    {
        int fa=read();LL d=read();
        ins(fa,i,d);
    }
    v[1]=0;dfs(1);
    DFS(1);
    for(int i=1;i<=n;i++)printf("%d\n",ans[i]);
}