Serval and Rooted Tree CodeForces - 1153D 树形dp

题目链接：点击查看

题意：n个节点的树，每个节点可以得到孩子中的最大值或最小值，叶节点的权值分别为1 2 ... 叶节点数量，求节点1的最大值

题解：dp[i] 表示 i节点取最大值时，大于等于他的叶节点有几个，ans[i] 表示i节点取的最大值，当节点 i 取子代的最大值时，当然要从子代中取一个限制最少的，即大于等于他的最少的，取最小值时，就是要把所有的限制加起来了

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;
int n;
vector<int> v[N];
int vis[N],cnt,dp[N],ans[N];
void dfs(int u)
{
	if(v[u].size()==0)
	{
		dp[u]=1;
		ans[u]=cnt;
		return;
	}
	int minn=N,sum=0;
	for(int i=0;i<v[u].size();i++)
	{
		int to=v[u][i];
		dfs(to);
		if(vis[u]) minn=min(minn,dp[to]);
		else sum+=dp[to];
	}
//	cout<<u<<" "<<sum<<" "<<minn<<endl;
	if(vis[u]) dp[u]=minn;
	else dp[u]=sum;
	ans[u]=cnt-dp[u]+1;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&vis[i]);	
	int x;
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&x);
		v[x].push_back(i);
	}
	for(int i=1;i<=n;i++)
	{
		if(v[i].size()==0)
			cnt++;
	}
	dfs(1);
	printf("%d\n",ans[1]);
	return 0;
}