BZOJ3011 树上倍增

为什么很多人都拿左偏树来写这道题

好像没什么人写倍增

觉得倍增写起来更简单点

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e5+10;
const ll inf=1ll<<60;
int head[N],Next[N],ver[N],A[N],f[N][20],tot,t;
ll dis[N];
void add(int x,int y){
    ver[++tot]=y,Next[tot]=head[x],head[x]=tot;
}
void dfs1(int x,int y){
    f[x][0]=y;dis[x]=dis[x]+dis[y];
    for(int i=1;i<=t;++i)f[x][i]=f[f[x][i-1]][i-1];
    for(int i=head[x];i;i=Next[i]){
        dfs1(ver[i],x);
    }
}
int find(int x,ll k){
    for(int i=t;i>=0;--i){
        if(f[x][i]>0&&dis[x]-dis[f[x][i]]<=k)
            k-=dis[x]-dis[f[x][i]],x=f[x][i];
    }
    return x;
}
void dfs2(int x,int y){
    for(int i=head[x];i;i=Next[i]){
        dfs2(ver[i],x);
        A[x]+=A[ver[i]];
    }
}
int main(){
    //freopen("testdata.in","r",stdin);
    int n,x;ll k,y;cin>>n>>k;
    t=int(log(n)/log(2))+1;
    for(int i=2;i<=n;++i)scanf("%d%lld",&x,&y),add(x,i),dis[i]=y;
    dfs1(1,0);
    for(int i=1;i<=n;++i){
        x=find(i,k);
        A[i]++;
        A[f[x][0]]--;
    }
    dfs2(1,0);
    for(int i=1;i<=n;++i)printf("%d\n",A[i]);
}