Codeforces Round #551(div2) D.Serval and Rooted Tree（Dfs）

D. Serval and Rooted Tree


　
　　题意：给出树的大小n，给出树上每个点的状态（0或1）0代表此结点的值为所有子结点的最小值，1代表此结点的值为所有子结点的最大值。给出从2号结点开始到n号结点的父结点，（以1号结点为根结点）。求出根节点的最大值是多少。
　　数据范围：2 <= n <= 3e5。
　　涉及算法：Dfs。
　　思路：先提出最终答案的表达式：ans = cnt - y + 1。（y表示当前这个结点需要几个数就可以使其值最大，cnt表示共有几个底层结点）　　　对于最底层的结点而言，要让值最大只需要一个数就可以，所以所有底层结点的y都为1。对于其他结点而言，如果状态为1，则是min（所有子结点所需要的y），如果状态为0，则是 += 所有子结点所需要的y。因为状态为1时，所有子结点只要让其中一个的值最大就可以使此结点的值最大，状态为0时，要使此结点的值最大则需要将所有子结点的值从最大填到最小共结点个数个数，即当前的结点的值为n - 结点个数 + 1。
　　反思：没有清楚理解样例显现出来的规律，表达式无法写明。
AC代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 3e5;
const int inf = 0x3f3f3f;

int n;
bool a[maxn + 5];
vector <int> G[maxn + 5];

int Dfs(int x){
	if(G[x].size() == 0) return 1;
	if(a[x] == 1){
		int len = G[x].size();
		int ans = inf;
		for(int i = 0; i < len; i++){
			ans = min(ans, Dfs(G[x][i]));
		}
		return ans;
	}
	if(a[x] == 0){
		int len = G[x].size();
		int ans = 0;
		for(int i = 0; i < len; i++){
			ans += Dfs(G[x][i]);
		}
		return ans;
	}
}

int main(){
	scanf("%d", &n);
	for(int i = 1; i <= n; i++){
		scanf(" %d", &a[i]);
	}
	for(int i = 1; i <= n - 1; i++){
		int f;
		scanf(" %d", &f);
		G[f].push_back(i + 1);
	}
	int cnt = 0;
	for(int i = 2; i <= n; i++){
		if(G[i].size() == 0) cnt++;
	}
	int ans = cnt - Dfs(1) + 1;
	printf("%d\n", ans);
	return 0;
}