树形结构+贪心思维 黑白树

本文探讨了一种在有根树结构中进行最优染色的算法。目标是最少操作次数下将所有节点从白色变为黑色,通过计算节点间的距离和特定值k,优化染色路径。介绍了关键的数据结构和递归深度优先搜索策略。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目描述:

一棵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)

输出描述:

一个数表示最少操作次数


在这里插入图片描述
配合图片更好理解

因为第6个点必须涂色,所以涂了第6个点,第5.4个点也涂了,这个时候如果又从第3个点开始涂,明显是不优的,直接再涂第5个点就可以了;

所以只要维护每个结点子树的最大k值,还有每个结点可以往上涂的个数 f ,当f=0时,说明不能往上涂了,这时 f 直接更新为当前结点的 k 值;

代码:

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=100100;
const int M=50100;
const LL mod=1e9+7;
int n,head[N],cnt;
int k[N],f[N],ans;
struct Node{
	int to,nex;
}edge[N*2];
void add(int p,int q){
	edge[cnt].to=q,edge[cnt].nex=head[p],head[p]=cnt++;
}
void dfs(int p,int ft){
	for(int i=head[p];~i;i=edge[i].nex){
		int q=edge[i].to;
		if(q!=ft){
			dfs(q,p);
			f[p]=max(f[p],f[q]-1);
			k[p]=max(k[p],k[q]-1);
		}
	}
	if(f[p]==0){
		ans++;
		f[p]=k[p];
	}
}
int main(){
	memset(head,-1,sizeof(head));
	cin>>n;
	for(int i=2;i<=n;i++){
		int p;cin>>p;
		add(i,p),add(p,i);
	}
	for(int i=1;i<=n;i++) cin>>k[i];
	dfs(1,0);
	cout<<ans<<endl;
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值