【BZOJ】 4813: [Cqoi2017]小Q的棋盘

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=4813

---

暴力转移就好，考虑以某一个点为根的子树分为是否走回来两种情况

${f_{i,j}}$表示已点$i$为根的子树，走了$j$步之后回到点$i$最多能经过多少个点

${g_{i,j}}$表示已点$i$为根的子树，走了$j$步之后不管停在那个点最多能经过多少个点

写了个${n^{2}}$转移

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<map>
using namespace std;
#define llg int
#define maxn 210
#define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
vector<llg>a[maxn];
llg f[maxn][maxn],g[maxn][maxn],n,m;
inline llg getint()
{
    llg w=0,q=0; char c=getchar();
    while((c<'0' || c>'9') && c!='-') c=getchar();
    if (c=='-')  q=1, c=getchar(); while (c>='0' && c<='9') w=w*10+c-'0', c=getchar();
    return q ? -w : w;
}

void init()
{
    cin>>n>>m;
    for (llg i=1;i<n;i++)
    {
        llg x=getint()+1,y=getint()+1;
        a[x].push_back(y),a[y].push_back(x);
    }
}

void dp(llg x,llg fa)
{
    llg w=a[x].size(),v;
    f[x][0]=g[x][0]=1;
    for (llg e=0;e<w;e++)
    {
        v=a[x][e];
        if (v==fa) continue;
        dp(v,x);
        for (llg i=m;i>=1;i--)
        {
            for (llg j=0;j<i;j++)
            {
                if (i-j>=2)
                {
                    f[x][i]=max(f[x][i],f[v][j]+f[x][i-j-2]);
                    g[x][i]=max(g[x][i],f[v][j]+g[x][i-j-2]);
                }
                g[x][i]=max(g[x][i],g[v][j]+f[x][i-j-1]);
            }
        }
    }
    for (llg i=0;i<m;i++) f[x][i+1]=max(f[x][i+1],f[x][i]),g[x][i+1]=max(g[x][i+1],g[x][i]);
}

int main()
{
    yyj("dp");
    init();
    dp(1,0);
    cout<<g[1][m];
    return 0;
}