bzoj4813 [Cqoi2017]小Q的棋盘 树形dp

Description

---

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

V,N≤ 100, 0 ≤Ai,Bi<V

Solution

---

直接dp，f[i,j,0/1]表示i子树内走j步，是否回到i的答案，转移比较显然

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)

const int N=205;

struct edge {int y,next;} e[N*2];

int ls[N],edCnt,n,m;
int f[N][N][2];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void add_edge(int x,int y) {
	e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
	e[++edCnt]=(edge) {x,ls[y]}; ls[y]=edCnt;
}

void update(int &x,int v) {
	x=(x<v)?(v):(x);
}

void dfs(int now,int fa) {
	f[now][0][1]=f[now][0][0]=1;
	for (int i=ls[now];i;i=e[i].next) {
		if (e[i].y==fa) continue;
		dfs(e[i].y,now);
		/*drp(j,m,0) {
			rep(k,0,j-1) {
				update(f[now][j][0],f[now][j-k-1][1]+f[e[i].y][k][0]);
				if (j-k-2>=0) {
					update(f[now][j][0],f[now][j-k-2][0]+f[e[i].y][k][1]);
					update(f[now][j][1],f[now][j-k-2][1]+f[e[i].y][k][1]);
				}
			}
		}*/
		drp(j,m,0) rep(k,0,m) {
			update(f[now][j+k+1][0],f[now][j][1]+f[e[i].y][k][0]);
			update(f[now][j+k+2][0],f[now][j][0]+f[e[i].y][k][1]);
			update(f[now][j+k+2][1],f[now][j][1]+f[e[i].y][k][1]);
		}
	}
	rep(i,0,m-1) {
		update(f[now][i+1][0],f[now][i][0]);
		update(f[now][i+1][1],f[now][i][1]);
	}
}

int main(void) {
	n=read(),m=read();
	rep(i,2,n) {
		int x=read(),y=read();
		add_edge(x,y);
	}
	dfs(0,0);
	printf("%d\n", f[0][m][0]);
	return 0;
}