黑白树（牛客网+树形dp）

链接：https://ac.nowcoder.com/acm/problem/13249
来源：牛客网

题目描述
一棵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
我们从每个叶子节点往上回溯。
如果子树里面的点染不到这个点就ans++,然后将这个点的染色值传上去。
如果可以染到这个点，就更新父节点的染色值，然后记录当前节点染色到的最大值，并且返回。
代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int maxx=1e5+100;
struct node{
	int to;
	int next;
}e[maxx<<1];
int tot,head[maxx<<1],k[maxx],fa[maxx],n;
/*----------事前准备---------*/
inline void init()
{
	memset(head,-1,sizeof(head));
	tot=0;
}
inline void add(int u,int v)
{
	e[tot].to=v,e[tot].next=head[u],head[u]=tot++;
}
/*---------树形dp----------*/
inline int dfs(int u,int f,int &ans)
{
	fa[u]=f;
	int now=0;
	for(int i=head[u];i!=-1;i=e[i].next)
	{
		int to=e[i].to;
		if(to==f) continue;
		now=max(now,dfs(to,u,ans));
	}
	if(now<=1)//子树里面的点染不到当前的节点
	{
		ans++;
		return k[u];
	}
	k[f]=max(k[f],k[u]-1);//如果能够染到的话，就更新父亲节点的染色值，取最大值
	return now-1;
}

int main()
{
	int x;
	while(~scanf("%d",&n))
	{
		init();
		for(int i=2;i<=n;i++)
		{
			scanf("%d",&x);
			add(i,x);add(x,i);
		}
		for(int i=1;i<=n;i++) scanf("%d",&k[i]);
		int ans=0;
		dfs(1,0,ans);
		cout<<ans<<endl;
	}
	return 0;
}

努力加油a啊，(^o)/~