luogu 3698 [CQOI2017]小Q的棋盘 树形dp

Code:

#include <bits/stdc++.h>
#define N 107
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
int n,K,f[N][N],g[N][N],tmp[N][N];
vector<int>G[N];
void add(int u,int v)
{
    G[u].push_back(v);
}
void getmax(int &a,int b)
{
    if(b>a) a=b;
}
void dfs(int u,int ff)
{
    f[u][0]=tmp[u][0]=1;
    for(int i=0;i<G[u].size();++i)
    {
        int v=G[u][i];
        if(v==ff) continue;
        dfs(v,u);
        for(int j=K;j>=0;--j)
        {
            for(int d=0;d<=j;++d)
            {
                if(j-d-1>=0)
                {
                    getmax(tmp[u][j], f[u][j-d-1]+g[v][d]);          // 新走
                }
                if(j-d-2>=0) getmax(tmp[u][j],tmp[u][j-d-2]+f[v][d]);         // 以前走过
            }
        }
        for(int j=K;j>=0;--j)
        {
            for(int d=0;d<=j;++d)
                if(j-d-2>=0) getmax(f[u][j], f[u][j-d-2]+f[v][d]);
        }
    }
    for(int i=0;i<=K;++i) g[u][i]=tmp[u][i];
}
int main()
{
    int i,j;
    // setIO("input");
    scanf("%d%d",&n,&K);
    for(i=1;i<=n-1;++i)
    {
        int a,b;
        scanf("%d%d",&a,&b),add(a+1,b+1),add(b+1,a+1);
    }
    dfs(1,0);
    int ans=0;
    for(i=1;i<=K;++i) getmax(ans, g[1][i]);
    printf("%d\n",ans);
    return 0;
}