黑白树-----------------------------------------------树形dp

在这里插入图片描述
在这里插入图片描述
解析:
染色肯定自下往上染,所以我们要从叶子结点开始染色。
先看子树里的点能不能染到这个点,染不到就++ans,把这个点的k传上去
能染到的话要更新父节点的染色值(因为有可能儿子节点染色值大于本身,这会让答案更优)

染色值往上传递是要-1的
递推方程 k[fa]=max(k[fa],k[u]-1)

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10000;
vector<int> G[N];
int father[N];
int k[N],n,u;
int dfs(int u,int fa,int &ans)
{
	father[u]=fa;
	int now=0;
	for(auto j : G[u])
	{
		if(j==fa) continue;
		now=max(now,dfs(j,u,ans));
	}
	if(now<=1)
	{
		ans++;
		return k[u];
	}
	k[fa]=max(k[fa],k[u]-1);
	return now-1;
}
int main()
{
	while(~scanf("%d",&n))
	{
		int ans=0; 
		for(int i=1;i<=n;i++) G[i].clear();
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&u);
  		G[i].push_back(u);G[u].push_back(i);	
	}
	for(int i=1;i<=n;i++) scanf("%d",&k[i]);
	dfs(1,0,ans);
	cout<<ans<<endl;
	} 
 } 
发布了572 篇原创文章 · 获赞 14 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_43690454/article/details/105289856