GYM - 101147 J.Whistle's New Car

题意：

　　给出一颗有点权和边权的树。求每一个点u的子树中有多少点v，使得点v到点u的距离小于等于点v的权值。

题解：

　　对于每一个点，倍增的预处理出他的祖宗节点及距离。根据预处理的结果求出每个点能到的最远的祖宗节点。

　　设点u能到的最远祖宗节点为v。利用差分的思想在点tree[u]+1，点tree[v]-1。

　　最后每个点的答案就是子树的tree[]和。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int maxn = 5e5+;

int t, n, u, v, tree[maxn], ans[maxn];

ll c, x[maxn];

struct node {

    ll val; int nod;

};

node fa[][maxn], tmp;

vector<node> g[maxn];

void dfs(int u, int pre) {

    fa[][u].nod = pre;

    int len = g[u].size();

    for(int i = ; i < len; i++) {

        int v = g[u][i].nod;

        if(v==pre) {

            fa[][u].val = g[u][i].val;

            continue;

        }

        dfs(v, u);

    }

}

void dfs_double(int u, int pre) {

    int len = g[u].size();

    for(int i = ; i < len; i++) {

        int v = g[u][i].nod;

        if(v==pre) continue;

        dfs_double(v, u);

    }

    tree[u]++;

    ll tv = x[u];

    for(int k = ; k >= ; k--) {

        if(fa[k][u].val <= tv) {

            tv -= fa[k][u].val;

            u = fa[k][u].nod;

        }

    }

    tree[u]--;

}

void solve(int u, int pre) {

    int len = g[u].size();

    for(int i = ; i < len; i++) {

        int v = g[u][i].nod;

        if(v==pre) continue;

        solve(v, u);

    }

    for(int i = ; i < len; i++) {

        int v = g[u][i].nod;

        if(v==pre) continue;

        ans[u] += tree[v]+ans[v];

    }

}

int main() {

    freopen("car.in","r",stdin);

    scanf("%d", &t);

    while(t--) {

        scanf("%d", &n);

        memset(tree, , sizeof(tree));

        memset(ans, , sizeof(ans));

        for(int i = ; i <= n; i++) g[i].clear();

        for(int i = ; i <= n; i++) scanf("%lld", &x[i]);

        for(int i = ; i < n; i++) {

            scanf("%d%d%lld", &u, &v, &tmp.val);

            tmp.nod = v;

            g[u].push_back(tmp);

            tmp.nod = u;

            g[v].push_back(tmp);

        }

        dfs(, );

        for(int k = ; k < ; k++) {

            for(int v = ; v <= n; v++) {

                fa[k+][v].nod = fa[k][fa[k][v].nod].nod;

                fa[k+][v].val = fa[k][v].val+fa[k][fa[k][v].nod].val;

            }

        }

        dfs_double(, );

        solve(, );

        for(int i = ; i <= n; i++) {

            printf("%d", ans[i]);

            if(i==n) puts("");

            else printf(" ");

        }

    }

}