bzoj3011: [Usaco2012 Dec]Running Away From the Barn 可并堆(左偏树)

题意：给出以1号点为根的一棵有根树，问每个点的子树中与它距离小于l的点有多少个。n<=2e5,l<=1e18

随手抽了一道，以为是个水题，刷水不成反被刷。
一开始觉得这题不难，然后样例老过不去，然后看zyf2000的代码。。还是错的，最后直接拿她的交了一下，结果是错的。。一怒之下删掉重写。。
开始想的是倍增什么的，但是好像不好动态维护，而且复杂度好大。。
然后这就只能可并堆咯，堆维护每个子树内距离不超过L的点，从下往上走，走完一个子树就合并一波，然后动态把距离超过L的删掉就可以了。
很久没打左偏树，好丑。。
需要注意的是d[i]是左偏树的dist，dis[i]则是i到根的距离，刚好与左偏树的优先级是相同的。
记录lazy[x]来传递权值。f[x]表示x所在的堆的top。

#include<cstdio>#include<algorithm>#include<cstring>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fd(i,a,b) for(int i=a;i>=b;i--)using namespace std;const int N=4e5+5;typedef long long ll;int head[N],next[N],go[N],d[N];int ans[N],n;ll dis[N],lazy[N],val[N],L;int ls[N],rs[N],f[N],size[N],tot;inline void add(int x,int y,ll z){    go[++tot]=y;    val[tot]=z;    next[tot]=head[x];    head[x]=tot;}inline void solve(int x,ll v){    if (!x)return;    dis[x]+=v;    lazy[x]+=v;}inline void pushup(int x){    size[x]=size[ls[x]]+size[rs[x]]+1;}inline void pushdown(int x){    if (lazy[x])solve(ls[x],lazy[x]),solve(rs[x],lazy[x]),lazy[x]=0;}inline 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;}inline void pop(int x){    pushdown(f[x]);    f[x]=merge(ls[f[x]],rs[f[x]]);}inline void dfs(int x,int fa){    f[x]=x,d[x]=size[x]=1;    for(int i=head[x];i;i=next[i])    {        int v=go[i];        if (v!=fa)        {            dfs(v,x);            solve(f[v],val[i]);            while (dis[f[v]]>L)pop(v);            f[x]=merge(f[x],f[v]);        }    }    ans[x]=size[f[x]];}int main(){    scanf("%d%lld",&n,&L);    fo(i,2,n)    {        int x;        ll z;        scanf("%d%lld",&x,&z);        add(x,i,z);        add(i,x,z);    }    dfs(1,0);    fo(i,1,n)printf("%d\n",ans[i]);}