Serval and Rooted Tree(树形dp）

LINK

题目：

样例：

样例解释：

题意：

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

思路：

想复杂了，看了一下大神们的题解，只能说：nb哇！！！
运用树形dp。
dp[i]：以i为根，能到达第dp[i]大的值；
在每个叶子节点上dp[叶子]=1（只要叶子自己本身赋值最大即最大值啦）
对于非叶子节点，首先看它取最小值还是最大值；
1.取最大值时：此位置所需叶子数量即它的子节点中最小所需数量即可；
2.取最小值时：此位置所需叶子数量即它的子节点所需叶子数量相加之和；(这时子节点所用的值相冲（每个值只能使用一次），所以能到达最大值的位数为：$dp[dad]=\sum dp[son]$)
转换为代码：

	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;
}