树上贪心即可解决此题:
dfs的时候每往下走一步让对应结点的k值-1,因为要覆盖到同一个点,孩子比父亲会远一个距离,当无法覆盖的时候(k值被减到0了),新开辟一个结点(需要染色),然后k值又重新开始减。
#include <bits/stdc++.h>
using namespace std;
const int N=100100;
int k[N];
vector <int> G[N];
int ans=0;
int dfs(int root,int pre)
{
int sum=0;
for(int i=0;i<G[root].size();i++)
{
int to=G[root][i];
if(to==pre) continue;
sum=max(sum,dfs(to,root));
}
if(!sum)
{
ans++;
return k[root]-1;
}
k[pre]=max(k[pre],k[root]-1);
return sum-1;
}
int main()
{
int n;
scanf("%d",&n);
for(int i=2;i<=n;i++)
{
int fa;
scanf("%d",&fa);
G[fa].push_back(i);
}
for(int i=1;i<=n;i++) scanf("%d",&k[i]);
dfs(1,0);
printf("%d\n",ans);
} 


京公网安备 11010502036488号