luoguP3066 Running Away From the Barn G 滑动窗口 树上差分

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

所以我们边深搜边维护一个递归栈，然后不断在栈内滑动，保持窗口内边的权值和小于k。从而得出每个点会给多少个点的答案做贡献。然后使用树上差分维护。

#include <cstdio>
using namespace std;
const int MAXN = 210000;
typedef long long ll;
int head[MAXN],to[MAXN],nxt[MAXN],dep[MAXN],fa[MAXN],pot[MAXN],tot[MAXN];
ll val[MAXN],sum[MAXN];
int cnt,n;
ll k;
void add(int x,int y,ll v)
{
    nxt[++cnt] = head[x];
    to[cnt] = y;
    head[x] = cnt;
    val[cnt] = v;
}
void change(int rt,int x,int y)
{
    tot[rt]--;
    tot[fa[rt]]--;
    tot[x]++;
    tot[y]++;
}
void dfs(int x,ll now,int hd)
{
    pot[dep[x]] = x;
    ll tnow = now;
    int thd = hd;
    change(pot[hd],pot[hd],pot[dep[x]]);
    for (int i = head[x];i;i = nxt[i])
    {
        dep[to[i]] = dep[x] + 1;
        fa[to[i]] = x;
        sum[dep[x]] = val[i];
        while (now + val[i] > k)
            now -= sum[hd++];
        now += val[i];
        dfs(to[i],now,hd);
        now = tnow;
        hd = thd;
    }
}
void calc(int x)
{
    for (int i = head[x];i;i = nxt[i])
    {
        calc(to[i]);
        tot[x] += tot[to[i]];
    }
}
int main()
{
    scanf("%d%lld",&n,&k);
    int tx;
    ll tl;
    for (int i = 2;i <= n;i++)
    {
        scanf("%d%lld",&tx,&tl);
        add(tx,i,tl);
    }
    dep[1] = 1;
    dfs(1,0,1);
    calc(1);
    for (int i = 1;i <= n;i++)
        printf("%d
",tot[i]); 
    return 0;
}