cf 1153D-Serval and Rooted Tree（思维，树形dp）

传送门

题意

给定一棵具有 $n$ 个结点的数，每个结点都被标有 $max$ 或 $min$ 标记，分别表示该点的值将取其所有儿子结点中的最大值或最小值（对于叶子结点可忽略该标记）。该树共有 $k$ 个叶子结点，可以对这 $k$ 个叶子进行 $[1,k]$ 范围内的赋值，求最后根节点可取的最大值。

本题中并未要求精确到所有叶子的构造方案，因此可以考虑从 $max$ 和 $min$ 的性质入手。对于任一为 $max$ 的结点，若其的儿子有 $i$ 个，则他最终赋到的值将是这 $i$ 个结点中第 $1$ 大的值，而若标记为 $min$ ，则将赋为第 $i$ 大的值。根据这个特点，可以考虑将叶节点的赋值转化为用 $[1,k]$ 中第几大的值进行标记。例如：赋值为 $k-1$ 的标记为 $2$。

所以对于所有结点，只需要贪心的考虑把最优策略放到该点及以下的叶子中，并考虑该点在最优策略下能够取得的最大值。容易想到，对于某个标记为 $max$ 的结点（若为叶子结点则标记直接为 $1$ ），它的标记将为所有儿子中标记的最小值，而对于标记为 $min$ 的结点，其标记应为所有儿子标记的加和（该原理解释待补图）。由此进行树形 $dp$ 即可。

#include <bits/stdc++.h>
#define int long long
using namespace std;

const int inf=0x3f3f3f3f;
const int N=3e5+10;
int tag[N],f[N];
vector<int> h[N];

void dfs(int u){
	int minn=0,maxx=inf;
	if(h[u].size()==0){
		f[u]=1;
		return ;
	}
	
	for(auto x:h[u]){
		dfs(x);
		maxx=min(maxx,f[x]);
		minn+=f[x];
	}
	
	if(tag[u])
		f[u]=maxx;
	else
		f[u]=minn;
}

signed main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>tag[i];
    }
	for(int i=2;i<=n;i++){
		int x;
		cin>>x;
		h[x].push_back(i);
	}
	
	int k=0;
	for(int i=1;i<=n;i++){
		if(h[i].size()==0)
			k++;
	}
	
	dfs(1);
	int ans=k-f[1]+1;
	cout<<ans;
    return 0;
}