树上贪心即可解决此题:
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); }