牛客 每日一题 11 黑白树 题解（贪心）

题目链接

题目思路

前言

本来以为是直接从下往上dfs一遍，边处理边向上。后面发现是错误思想。
反例
叶子节点6一定得选，这个时候5和4都在他的覆盖范围内，如果都不选，3没被覆盖，要选，2要选1要选，一共选了4个点，但是事实上我们选6和5就行了。

正文

我们还是看上图：当我们按照能覆盖上就不选的方法从下往上选的时候，发现了3没有被下方的点覆盖，我们要做的事情其实是找个点（可能是自己可能是下方的点）覆盖了它，而不仅仅是选择它，在这个例子中显然选择5比选择3更好——这两种选择都能满足覆盖3，但是选择5能到覆盖更上方的点。也就说说，我们真正的策略应该是，当一个点没有被下方已经选了的点覆盖到的时候，我们选择一个它或者它下方的能向上覆盖的最远的点。另外，其实我们也并不关心这个点选的哪个，只需要知道往上能覆盖多远就行了，这个值其实可以直接维护到k数组里面，即我认为由于图中5号点的存在，在dfs返回上方的过程中，我们就可以认为k[4]值是9，k[3]是8（选6来替代选3/4）……

我们用一个f[x]表示在x的子树都被覆盖的当前选择下，最远能往x的上方覆盖多远，f[x] = max(f[son] - 1) ，son是x的儿子。（如果这个点还没选，那么它不会用来更新f的值的，即dfs（4）的时候f[4]是等于0的）

如果一个点所有的儿子都不能向上覆盖到它了，那就在找个点及其子树里面选一个能往上覆盖最远的把它覆盖了，选择的点数+1，且f值被k值更新，如果子树还能覆盖到它，就不用在这里选了（在考虑子树被覆盖的问题的时候就已经选过了。）

代码

个人觉得代码尽管比较短，但是有点难理解自己也有点模糊

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
int x,n,cnt,head[maxn],k[maxn],f[maxn],ans;
struct node{
	int to,next;
}e[maxn];//单向边 
void add(int u,int v){
	e[++cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}
void dfs(int son,int fa){
	for(int i=head[son];i;i=e[i].next){
		if(e[i].to!=fa){
			dfs(e[i].to,son);
			f[son]=max(f[son],f[e[i].to]-1);//维护f值——儿子的f值-1之后的最大值
			k[son]=max(k[son],k[e[i].to]-1);//维护k值——儿子的k值-1和自己的k值的最大值
		}
	}
	if(f[son]==0){//下面的点都覆盖不到它了——选他自己
		ans++;
		f[son]=k[son];
	}
}
int main(){
	scanf("%d",&n);
	for(int i=2;i<=n;i++){
		scanf("%d",&x);
		add(x,i);//注意加边的顺序 
	}
	for(int i=1;i<=n;i++){
		scanf("%d",&k[i]);
	}
	dfs(1,1);
	printf("%d\n",ans);
	return 0;
} 

参考链接https://ac.nowcoder.com/discuss/401669