codeforces 1153D Serval and Rooted Tree

https://codeforces.com/problemset/problem/1153/D

太菜了，1800分的题没点思路。

看了别人代码，我们可以用dp来实现，dp[i]表示第i个点为根节点的子树，他需要消耗的叶子节点是多少个。

一个点需要消耗dp[i]个点，意思也就是说给他分配k个数，只有dp[i]个数会消耗实际数字个数，其他数字随便分。

那么答案就是k-dp[1]+1.

叶子节点自身dp[u]=1;

如果u取所有儿子v中最小的，那么dp[u]=dp[v1]+dp[v2]....因为他要从所有中取最小的，那么所有儿子都得消耗个数

如果是取max，那么只要让一个儿子最大就行了，那么久从dp[v]中取最小的。

#include<bits/stdc++.h>
#define maxl 300010
using namespace std;

int n,cnt,ans;
int a[maxl],ehead[maxl],f[maxl],dp[maxl];
struct ed
{
	int to,nxt;
}e[maxl<<1];

inline void add(int u,int v)
{
	e[++cnt].to=v;e[cnt].nxt=ehead[u];ehead[u]=cnt;
}

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&f[i]);
		add(f[i],i);
	}
}

inline void mainwork()
{
	for(int u=n;u>=1;u--)
	{
		if(ehead[u]==0)
		{
			dp[u]=1;ans++;
			continue;
		}
		int v;
		if(a[u]==0)
			for(int i=ehead[u];i;i=e[i].nxt)
			{
				v=e[i].to;
				dp[u]+=dp[v];
			}
		else
		{
			dp[u]=1e6;
			for(int i=ehead[u];i;i=e[i].nxt)
			{
				v=e[i].to;
				dp[u]=min(dp[u],dp[v]);
			}
		}
	}
}

inline void print()
{
	ans=ans-dp[1]+1;
	printf("%d\n",ans);
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}