【牛客网】黑白树（树型dp）

最新推荐文章于 2024-10-01 18:52:56 发布

准图灵奖得主

最新推荐文章于 2024-10-01 18:52:56 发布

阅读量178

点赞数

分类专栏：算法刷题文章标签：动态规划树形DP 链式前向星题解 c++

本文为博主原创文章，未经允许不得转载。如有问题，欢迎指正！

本文链接：https://blog.csdn.net/IAMLSL/article/details/114242941

版权

算法刷题专栏收录该内容

67 篇文章 5 订阅

订阅专栏

题目描述：

在这里插入图片描述

解题思路：

AC代码：

#include<bits/stdc++.h>
     using namespace std;
     const int maxn=1e5+100;
     struct edge{
     int u;
     int v;
     int next;
     }e[2*maxn];
     int cnt=0,head[2*maxn],vis[maxn],s[maxn],k[maxn],to[maxn];
     void Insert(int u,int v){
     cnt++;
     e[cnt].v=v;
     e[cnt].next=head[u];
     head[u]=cnt;
     }
     int Max(int x,int y){
     return x>y ? x : y;
     }
     int ans=0;
     void  treedp(int cur){
      vis[cur]=1;
      if(!to[cur]){
      s[cur]=k[cur];
      ans++;
      return;
      }
      for(int i=head[cur];i>=0;i=e[i].next){
      int v=e[i].v;
      if(!vis[v]){
      treedp(v);
      k[cur]=Max(k[cur],k[v]-1);
      s[cur]=Max(s[cur],s[v]-1);
      }
      }
     if(!s[cur]){
     ans++;
     s[cur]=k[cur];
     }
     }
     int main(){
     int n;
     cin>>n;
     memset(head,-1,sizeof(head));
     for(int i=2;i<=n;i++){
     int fa;
     cin>>fa;
     Insert(fa,i);
     Insert(i,fa);
      to[fa]++;
     }
     for(int i=1;i<=n;i++)
          cin>>k[i];
     treedp(1);
     cout<<ans<<endl;
     return 0;
      }