bzoj 4813: [Cqoi2017]小Q的棋盘【树形dp】

这么简单的dp我怎么没想到x2

f为从这个点出发后回到这个点最多能走过的点，g为从这个点出发后不回到这个点最多能走过的点，注意g有两种转移：g[u][k]=max(g[u][k],f[u][k-j-1]+g[e[i].to][j])是在e[i].to这个子树前走了一棵子树再回来，g[u][k]=max(g[u][k],g[u][k-j-2]+f[e[i].to][j])是走了e[i].to的一棵子树之后回到e[i].to再走另一棵

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=105;
int n,m,h[N],cnt,f[N][N],g[N][N],ans;
struct qwe
{
	int ne,to;
}e[N<<1];
void add(int u,int v)
{
	cnt++;
	e[cnt].ne=h[u];
	e[cnt].to=v;
	h[u]=cnt;
}
void dp(int u,int fa)
{
	f[u][0]=g[u][0]=1;
	for(int i=h[u];i;i=e[i].ne)
		if(e[i].to!=fa)
		{
			dp(e[i].to,u);
			for(int k=m;k>=1;k--)
				for(int j=0;j<k;j++)
				{
					if(k-j-2>=0)
						f[u][k]=max(f[u][k],f[u][k-j-2]+f[e[i].to][j]);
					g[u][k]=max(g[u][k],f[u][k-j-1]+g[e[i].to][j]);
					if(k-j-2>=0)
						g[u][k]=max(g[u][k],g[u][k-j-2]+f[e[i].to][j]);
				}
		}
	for(int i=0;i<m;i++)
	{
		f[u][i+1]=max(f[u][i],f[u][i+1]);
		g[u][i+1]=max(g[u][i],g[u][i+1]);
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x+1,y+1);
		add(y+1,x+1);
	}
	dp(1,1);
	printf("%d\n",g[1][m]);
	return 0;
}