CF1065F Up and Down the Tree

题意:给你一棵树,你起点在1,1也是根节点,你每次可以选择去你子树的某个叶子节点,也可以选择,从叶子节点返回距离不超过k的一个根,也就是说,你从1开始,向下跳,选择一个叶子(就是没有子树的节点),然后可以选择一个距离小于等于k的点跳回去,然后继续跳下去再跳上来,问你最多能去多少个叶子节点,n<=1e6

题解

大概的意思就是树形dp，$h[u]$表示$u$点到最近的叶子结点的距离，$dp[u]$表示$u$点最多能到的叶子的数量，$a[u]$表示从$u$跳下去又能跳回来的最多的点数

然后大概的思路就是从下往上考虑，先把子树里跳了能回来的节点全都跳一遍，然后再跳下去永远不回来

然后似乎是$h[u]>=k$的时候把$a[u]$设为$0$，因为它对父亲已经不会有贡献了

我的理解的话，就是把子树里所有跳下去能回来的节点单独记录，然后判断一下跳下去就回不来的方法哪一个更优，然后能回来的跳完之后就跳下去永远不会来就好了

//minamoto
#include<bits/stdc++.h>
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
inline int read(){
    #define num ch-'0'
    char ch;bool flag=0;int res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
const int N=1e6+5,inf=0x3f3f3f3f;
int head[N],Next[N],ver[N],tot;
inline void add(int u,int v){
    ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
}
int dp[N],a[N],h[N],n,k;
void dfs(int u){
    h[u]=inf;
    for(int i=head[u];i;i=Next[i]){
        int v=ver[i];dfs(v);
        cmin(h[u],h[v]+1),a[u]+=a[v],cmax(dp[u],dp[v]-a[v]);
    }
    dp[u]+=a[u];
    if(h[u]==inf) dp[u]=a[u]=1,h[u]=0;
    if(h[u]>=k) a[u]=0;
}
int main(){
//    freopen("testdata.in","r",stdin);
    n=read(),k=read();
    for(int i=2,fa;i<=n;++i)
    fa=read(),add(fa,i);
    dfs(1);
    printf("%d
",dp[1]);
    return 0;
}