Codeforces1065F Up and Down the Tree 【树形DP】

推荐一道联赛练习题。

题目分析：

你考虑进入一个子树就可能上不来了，如果上得来的话就把能上来的全捡完然后走一个上不来的，所以这就是个基本的DP套路。

代码：

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

 const int maxn = ;

 int n,k,fa[maxn];
 vector <int> g[maxn];

 int dep[maxn],minn[maxn];

 int f[maxn],d[maxn];

 void read(){
     scanf("%d%d",&n,&k);
     for(int i=;i<=n;i++) {scanf("%d",&fa[i]);g[fa[i]].push_back(i);}
 }

 void dfs(int now,int dp){
     dep[now] = dp;
     if(g[now].size() == ){minn[now] = dp; return;}
     minn[now] = 1e8;
     for(int i=;i<g[now].size();i++){
     dfs(g[now][i],dp+);
     minn[now] = min(minn[now],minn[g[now][i]]);
     }
 }

 void dfs2(int now){
     if(g[now].size() == ){
     f[now] = d[now] = ;
     return;
     }
     for(int i=;i<g[now].size();i++) dfs2(g[now][i]);
     for(int i=;i<g[now].size();i++)
     if(minn[g[now][i]] - dep[now] <= k) d[now] += d[g[now][i]];
     for(int i=;i<g[now].size();i++){
     if(minn[g[now][i]] - dep[now] <= k)
         f[now] = max(f[now],d[now]-d[g[now][i]]+f[g[now][i]]);
     else f[now] = max(f[now],d[now]+f[g[now][i]]);
     }
 }

 void work(){
     dfs(,);
     dfs2(); // dp
     printf("%d\n",f[]);
 }

 int main(){
     read();
     work();
     return ;
 }