黑白树（DFS、贪心）

题目传送门

题目描述

一棵n个点的有根树，1号点为根，相邻的两个节点之间的距离为1。树上每个节点i对应一个值k[i]。每个点都有一个颜色，初始的时候所有点都是白色的。
你需要通过一系列操作使得最终每个点变成黑色。每次操作需要选择一个节点i，i必须是白色的，然后i到根的链上（包括节点i与根）所有与节点i距离小于k[i]的点都会变黑，已经是黑的点保持为黑。问最少使用几次操作能把整棵树变黑。

输入描述:

第一行一个整数n (1 ≤ n ≤ 10^5)
接下来n-1行，每行一个整数，依次为2号点到n号点父亲的编号。
最后一行n个整数为k[i] (1 ≤ k[i] ≤ 10^5)

样例解释:
对节点3操作，导致节点2与节点3变黑
对节点4操作，导致节点4变黑
对节点1操作，导致节点1变黑

输出描述:

一个数表示最少操作次数

示例1

输入

4
1
2
1
1 2 2 1

输出

3

分析：我们先读懂题目意思，若对某一白节点i进行染色，那么从节点i到根的链上所有与节点i距离小于k[i]的点都会变黑（说明染色方向单独朝向根），那么无论如何都必须对树的所有叶子节点染色，然后再根据每个节点权值k考虑是否对剩下的父节点染色，直到整棵树的节点全部变为黑节点，并且操作次数最少。

       但我们并不需急于先给叶子节点染色，我们可以先从根节点开始深搜每个节点的子节点，记录每个节点的深度。虽然我们知道每个节点的k值，但由于子节点和父节点处于不同深度，直接比较二者的k值的话不能确定哪个点是最优点。试想，我们可以把根节点看作基准，那么 该点的深度-该点的k值+1 就表示这个节点向上染色最多能够覆盖到的f深度，f深度越小说明该节点覆盖范围越大（相对）。这样的话我们运用贪心策略通过f深度来比较父节点和所有子节点哪个覆盖范围更大了。若子节点覆盖范围大，则同时更新父节点的f深度（父节点也被子节点涂色了，要继承子节点的光荣传统）；若子节点能覆盖父节点，但f深度比父节点小，则父节点的f深度不变；若子节点不能不能覆盖父节点，则父节点要额外进行一次涂色。最终返回到根节点，整棵树被涂色完毕。

以下是三种情况：

                                                                           

首先，4，5，6分别为叶子节点，没有子节点为它染色，故需要自己染色。再看他们的覆盖范围，4,5,6节点的深度为4，k值分别为3,2,1，f深度分别为2,3,4，可见在这一层中，4节点的覆盖范围最大，能覆盖到深度2的地方。在上一层节点3 k值为1，f深度为3，没有子节点4覆盖范围大，根据贪心策略被子节点染色并继承子节点的f深度。

                                                                           

4,5,6节点的深度为4，k值分别为2,2,1，f深度分别为3,3,4，可见在这一层中，4，5节点的覆盖范围最大，都能覆盖到深度3的地方。在上一层节点3 k值为1，f深度为3，也能被子节点覆盖，根据贪心策略被子节点染色并继承子节点的f深度

                                                                           

4,5,6节点的深度为4，k值分别为1,1,1，f深度分别为3,3,3，可见在这一层中，4，5，6节点的覆盖范围一样，只能覆盖到深度3的地方。在上一层节点3 k值为1，f深度为2，子节点不能覆盖父节点，根据贪心策略这时父节点要进行染色

#include<cstdio>
#include<vector>
#include<iostream>
using namespace std;
const int M = 1e5+5;
const int inf = 0x3f3f3f;
int n,f[M],p[M],k[M],deep[M],t,ans;
vector<int>e[M];
void dfs(int x){
    f[x] = deep[x]-k[x]+1;   //f[x]表示每个节点向上染色最多能够覆盖到的深度，
                             //深度越小说明该节点的染色范围越大
    p[x] = inf;
    for(int i=0;i<e[x].size();i++){ //遍历一个节点的下一深度的所有节点
        deep[e[x][i]] = deep[x]+1;   //统计每个节点的深度
        dfs(e[x][i]);    //深搜子节点的
        f[x] = min(f[x],f[e[x][i]]); //记录每个节点的最优染色范围
        p[x] = min(p[x],p[e[x][i]]); //记录每个节点的实际染色范围，用以判断是否需要染色
    }
    if(p[x]>deep[x]){   //对于叶子节点或子节点的p深度无法覆盖该节点的，该节点涂色
        ans++;
        p[x] = f[x];
    }
}
int main(){
    scanf("%d",&n);
    for(int i=2;i<=n;i++){
        scanf("%d",&t);
        e[t].push_back(i);  //统计每个节点的子节点
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&k[i]);  //统计每个节点的权值
    }
    deep[1] = 1;
    dfs(1);
    printf("%d\n",ans);

}
/*
4
1
2
1
1 2 2 1
*/