BZOJ3011: [Usaco2012 Dec]Running Away From the Barn 可并堆

题目链接：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]);
}