Codeforces Round #551 (Div. 2) D. Serval and Rooted Tree(树形DP)(好题*)

题目链接
在这里插入图片描述
在这里插入图片描述
题意:n个节点的树,叶子可以任意分别给1~k(k为叶子个数)的值,每个节点有操作min,max分别取其子树的最小和最大的值;

问根节点1最大多少;
思路:这个题不看题解根本不会往这方面想,这个思路很清奇,dp【u】代表u这个节点在它的叶子节点中排第几大,我们先来一种最简单的情况,dp【2】=3,dp【3】=4,总数是10,2和3的父亲是1,假如1取min的话,dp【1】=dp【2】+dp【3】=7,也就是它在它的10个叶子节点里排第7大(也就是3),如果1取max的话dp【1】=min(dp【2】,dp【3】)=3,也就是第3大(结果为7),是不是觉得很神奇?反正我是想不到QAQ。至于为什么min的话是累加,max是取最小,这个可以自己思考看看,以min为例,我已经知道子树的dp值,我们要取他们的最小同时又得保证每个子树的叶子的值互不相同只能累加

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e5+5;
vector<int>g[maxn];
int n,t,num,dp[maxn],a[maxn];
void dfs(int x)
{
	if(g[x].size()==0)
	{
		dp[x]=1;num++;return ;
	}
	dp[x]=a[x]?0x3f3f3f3f:0;
	for(int to:g[x])
	{
		dfs(to);
		if(a[x]) dp[x]=min(dp[x],dp[to]);
		else dp[x]+=dp[to];
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]);
	for(int i=2;i<=n;++i)
	{
		scanf("%d",&t);
		g[t].push_back(i);
	}
	dfs(1);
	printf("%d\n",num-dp[1]+1);
}
发布了391 篇原创文章 · 获赞 1 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_42479630/article/details/105458556