0
点赞
收藏
分享

微信扫一扫

Serval and Rooted Tree(树形dp)

月半小夜曲_ 2022-03-20 阅读 48

LINK

题目:

在这里插入图片描述

样例:

在这里插入图片描述

样例解释:

在这里插入图片描述
在这里插入图片描述

题意:

现有一棵树(根节点为1),树上每个结点上有一个flag值
(flag=0 :表示这个点的权值是所有子节点权值中的最小值;flag=1:表示这个点的权值是所有子节点权值中的最大值)。
有k个叶子节点,可以给每一个叶子节点安排1——k中的值且每个值只能放置一次。
问:根节点值的最大值是多少。

思路:

想复杂了,看了一下大神们的题解,只能说:nb哇!!!
运用树形dp。
dp[i]:i结点值最大时需要叶子的数量;
在每个叶子节点上dp[叶子]=1(只要叶子自己本身赋值最大即最大值啦)
对于非叶子节点,首先看它取最小值还是最大值;
1.取最大值时:此位置所需叶子数量即它的子节点中最小所需数量即可;
2.取最小值时:此位置所需叶子数量即它的子节点所需叶子数量相加之和;
转换为代码:

	if(cao[x]){//max
		dp[x]=0x3f3f3f3f;
		for(int i:g[x]){
			dfs(i);
			dp[x]=min(dp[x],dp[i]);
		}
	}else{
		for(int i:g[x]){
			dfs(i);
			dp[x]+=dp[i];
		}
	}

这样,根节点(1)能到达的最大值为 :叶子节点数目-dp[1]+1

代码:

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
//#define int long long
const int N=3e5+10;
const int nn=1e6+10;
const ll mod=1e9+7;
#define eps 1e-8

vector<int>g[N];
int cao[N],dp[N];
int num;//叶子数目 

void dfs(int x){//0:min  1:max
	if(g[x].size()==0){
		dp[x]=1;num++;return ;
	}
	if(cao[x]){//max
		dp[x]=0x3f3f3f3f;
		for(int i:g[x]){
			dfs(i);
			dp[x]=min(dp[x],dp[i]);
		}
	}else{
		for(int i:g[x]){
			dfs(i);
			dp[x]+=dp[i];
		}
	}
	
}
signed main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int n,x;cin>>n;
	for(int i=1;i<=n;i++)cin>>cao[i];
	for(int i=2;i<=n;i++){
		cin>>x;
		g[x].push_back(i);
	}
	dfs(1);
	cout<<num-dp[1]+1<<endl;
	return 0;
}
举报

相关推荐

0 条评论