【CQOI2017】bzoj4813 小Q的棋盘

树形dp，维护$f[i][j]$从点$i$往下走$j$再回来经过的最多点，$g[i][j]$从点$i$往下走$j$不用回来经过的最多点，不难转移。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int fir[110],ne[210],to[210],
f[110][110],g[110][110],
n,m;
void add(int num,int u,int v)
{
    ne[num]=fir[u];
    fir[u]=num;
    to[num]=v;
}
void dfs(int u,int fa)
{
    int v;
    for (int i=0;i<=m;i++) f[u][i]=g[u][i]=1;
    for (int i=fir[u];i;i=ne[i])
        if ((v=to[i])!=fa)
        {
            dfs(v,u);
            for (int j=m;j;j--)
                for (int k=1;k<=m&&k<=j;k++)
                {
                    f[u][j]=max(f[u][j],f[v][k-1]+g[u][j-k]);
                    if (k>=2)
                    {
                        f[u][j]=max(f[u][j],g[v][k-2]+f[u][j-k]);
                        g[u][j]=max(g[u][j],g[v][k-2]+g[u][j-k]);
                    }
                }
        }
}
int main()
{
    freopen("chessboard.in","r",stdin);
    freopen("chessboard.out","w",stdout);
    int u,v;
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        u++;
        v++;
        add(i<<1,u,v);
        add(i<<1|1,v,u);
    }
    dfs(1,-1);
    printf("%d\n",f[1][m]);
}