bzoj 4813 [Cqoi2017]小Q的棋盘

Description

小Q正在设计一种棋类游戏。在小Q设计的游戏中，棋子可以放在棋盘上的格点中。某些格点之间有连线，棋子只能
在有连线的格点之间移动。整个棋盘上共有V个格点，编号为0,1,2…,V-1，它们是连通的，也就是说棋子从任意格
点出发，总能到达所有的格点。小Q在设计棋盘时，还保证棋子从一个格点移动到另外任一格点的路径是唯一的。
小Q现在想知道，当棋子从格点0出发，移动N步最多能经过多少格点。格点可以重复经过多次，但不重复计数。
Input

第一行包含2个正整数V,N，其中V表示格点总数，N表示移动步数。
接下来V-1行，每行两个数Ai,Bi,表示编号为Ai,Bi的两个格点之间有连线。
V,N≤ 100, 0 ≤Ai,Bi < V
Output

输出一行一个整数，表示最多经过的格点数量。
Sample Input

5 2

1 0

2 1

3 2

4 3
Sample Output

3

从格点 0 出发移动 2 步。经过 0, 1, 2 这 3 个格点。

---

【分析】
树形dp
每个节点两种状态：0下去了没回来，1下去又回来了
三种转移竟然只写了两种我简直zz

---

【代码】

//cqoi 小Q的棋盘 
#include<iostream>
#include<cstring>
#include<cstdio>
#define ll long long
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=105;
int head[mxn],dp[mxn][mxn][2];   //是否回到当前节点 
int n,m,cnt,ans;  //n个节点,m步 
struct edge {int to,next;} f[mxn]; 
inline void add(int u,int v)
{
    f[++cnt].to=v,f[cnt].next=head[u],head[u]=cnt;
}
inline void dfs(int u,int fa)
{
    int i,j,v,k;
    dp[u][0][1]=dp[u][0][0]=1;
    for(i=head[u];i;i=f[i].next)
    {
        v=f[i].to;
        if(v!=fa) dfs(v,u);
        else continue;
        for(j=m;j>=1;j--)
          fo(k,0,j-1) 
          {
              if(j-k-2>=0)
                dp[u][j][0]=max(dp[u][j][0],dp[v][k][1]+max(dp[u][j-k-2][1],dp[u][j-k-2][0]));
              if(j-k-1>=0)
                dp[u][j][0]=max(dp[u][j][0],max(dp[v][k][0],dp[v][k][1])+dp[u][j-k-1][1]);
              if(j-k-2>=0)
                dp[u][j][1]=max(dp[u][j][1],dp[v][k][1]+dp[u][j-k-2][1]);
              ans=max(ans,max(dp[u][j][0],dp[u][j][1]));
          }
    }
}
int main()
{
    int i,j,u,v;
    scanf("%d%d",&n,&m);
    fo(i,2,n)
    {
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    dfs(0,-1);
    printf("%d\n",ans);
    return 0;
}
/* 
4 0
0 1
0 2
0 3
*/