Codeforces Round 551 (Div. 2)D. Serval and Rooted Tree（树形dp）

原题链接：D. Serval and Rooted Tree
题目大意：
给定一个有n个结点的以1为根的树，每个结点上有一个操作类型0或1，0表示min，1表示max，表示这个节点中的值应该分别等于其子节点中所有值的最大值或最小值。假设树上有k个叶节点，你可以将每个叶节点填上[1,k]的数字，且每个数字只使用一次，求根节点的最大值
数据范围：2≤n≤3⋅10⁵，1≤f_i≤i−1
解题思路：
考虑树形dp，从叶子节点开始分析，然后一步步往上推，最后得到根节点的状态，从而得到答案
首先分析题目，在一堆子节点中找到min和max，很难直接代数进去处理，可以考虑用排名来处理，设dp_u表示节点u在以u为根的子树中最小排名多少（最大的那个为第一名）
然后分类进行处理：
一.叶子节点：叶子节点为边界状态，没有子节点，全部设为1
二.如果这个节点是 max，则它的排名最小为它所有儿子的排名的最小值，最小值当我们给排名最小的儿子的子树分配最大的数据时得到
三.如果这个节点是min，若它的儿子i 的排名为k ，则先给它分配k个还没有被分配的最优的数据，剩下的数据随便分，得到它的排名最小为它所有儿子的排名之和
最后的结果为叶子节点个数 - 根节点排名(dp[1])+1

AC代码：

#include <bits/stdc++.h>
using namespace std;
#define int long long 
int a[300005],dp[300005];
vector<int> ed[300005];
int k=0,n;
void dfs(int x)
{
	//叶子节点
	if(ed[x].size()==0)
	{
		k++;
		dp[x]=1;
		return;
	}
	if(a[x])dp[x]=n;
	for(int i=0;i<ed[x].size();i++)
	{
		int p=ed[x][i];
		dfs(p);
		//max节点
		if(a[x])dp[x]=min(dp[x],dp[p]);
		//min节点
		else dp[x]+=dp[p];
	}
}
signed main(){
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=2;i<=n;i++)
    {
    	int fa;cin>>fa;
    	ed[fa].push_back(i);
	}
	dfs(1);
	cout<<k+1-dp[1]<<endl;
    return 0;
}