洛谷P14919 [GESP202512 六级] 路径覆盖

1.原题和原题链接

原题链接

P14919 [GESP202512 六级] 路径覆盖

题目背景

对应的选择、判断题：https://ti.luogu.com.cn/problemset/1202

题目描述

给定一棵有 $n$ 结点的有根树 $T$，结点依次以 $1,2,\dots ,n$ 编号，根结点编号为 $1$。方便起见，编号为 $i$ 的结点称为结点 $i$。

初始时 $T$ 中的结点均为白色。你需要将 $T$ 中的若干个结点染为黑色，使得所有叶子到根的路径上至少有一个黑色结点。将结点 $i$ 染为黑色需要代价 $c_i$，你需要在满足以上条件的情况下，最小化染色代价之和。

叶子是指 $T$ 中没有子结点的结点。

输入格式

第一行，一个正整数 $n$，表示结点数量。

第二行，$n-1$ 个正整数 $f_2,f_3,\dots ,f_n$，其中 $f_i$ 表示结点 $i$ 的父结点的编号，保证 $f_i<i$。

第三行，$n$ 个正整数 $c_1,c_2,\dots ,c_n$，其中 $c_i$ 表示将结点 $i$ 染为黑色所需的代价。

输出格式

一行，一个整数，表示在满足所有叶子到根的路径上至少有一个黑色结点的前提下，染色代价之和的最小值。

输入输出样例 #1

输入 #1

4
1 2 3
5 6 2 3

输出 #1

2

输入输出样例 #2

输入 #2

7
1 1 2 2 3 3
64 16 15 4 3 2 1

输出 #2

10

说明/提示

对于 $40\%$ 的测试点，保证 $2\le n\le 16$。

对于另外 $20\%$ 的测试点，保证 $f_i=i-1$。

对于所有测试点，保证 $2\le n\le 10^5$，$1\le c_i\le 10^9$。

2.主要思路

这道题要求在一棵有根树中，用最小的代价将某些节点染黑，使得每个叶子到根的路径上至少有一个黑色节点。我们可以使用动态规划从底部向上计算每个节点的最优解。对于每个节点x，有两种选择：要么直接染黑x（代价为c[x]），要么依靠子节点的染色结果（代价是子节点dp值之和）。dp[x]取这两种情况的最小值。所以通过dfs遍历，在回溯时完成状态转移。叶子节点的dp值就等于其染色代价，非叶子节点则比较自身代价和子节点代价的总和。最终根节点的dp值就是全局最优解。

3.AC代码

#include<bits/stdc++.h>
using namespace std;
vector<long long> v[100005];
long long c[100005],dp[100005],n;
void dfs(long long x){
	dp[x]=c[x];
	if(v[x].size()==0) return;
	long long sum=0;
	for(long long i:v[x]){
		dfs(i);
		sum+=dp[i];
	}
	dp[x]=min(c[x],sum);
}
main(){
	cin>>n;
	for(long long i=2;i<=n;i++){
		long long k;cin>>k;
		v[k].push_back(i);
	}
	for(long long i=1;i<=n;i++) cin>>c[i];
	dfs(1);
	cout<<dp[1];
	return 0;
}

以上就是本篇全部内容，感谢浏览！