--------------------------
http://www.cnblogs.com/rwy233/p/6689641.html
树形dp
f[i][j]表示在i这棵子树里走j步且最后要回到i最多能经过的节点数
g[i][j]则表示最后不用回到i
直接dfs每做完一个子树后转移一下就可以了
代码:
#include<cmath> #include<vector> #include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define CH c=getchar() #define For(i,x,y) for(int i=x;i<=y;++i) using namespace std; inline int read(){ bool f=0;char CH;for(;!isdigit(c);CH)if(c=='-')f=1; int x=0;for(;isdigit(c);CH) x=(x<<1)+(x<<3)+c-48;return f?-x:x; } const int N = 1e2+5; int f[N][N],g[N][N]; int n,m; int nxt[N<<1],to[N<<1],head[N],cnt; void add(int u,int v){ nxt[++cnt]=head[u];head[u]=cnt;to[cnt]=v; nxt[++cnt]=head[v];head[v]=cnt;to[cnt]=u; } void dfs(int u,int fat){ f[u][0]=g[u][0]=1; for(int i=head[u];i;i=nxt[i]){ int v=to[i]; if(v==fat) continue; dfs(v,u); for(int i=m;~i;--i){ for(int j=0;j<=i-1;++j){ if(i-j-2>=0)f[u][i]=max(f[u][i],f[v][j]+f[u][i-j-2]); g[u][i]=max(g[u][i],g[v][j]+f[u][i-j-1]); if(i-j-2>=0)g[u][i]=max(g[u][i],f[v][j]+g[u][i-j-2]); } } } for(int i=0;i<m;++i) f[u][i+1]=max(f[u][i],f[u][i+1]); for(int i=0;i<m;++i) g[u][i+1]=max(g[u][i],g[u][i+1]); } int main(){ // freopen("code.in","r",stdin); // freopen("code.out","w",stdout); n=read();m=read(); For(i,2,n){ int u=read(),v=read(); add(u,v); } dfs(0,0); printf("%d",g[0][m]); return 0; }转载于:https://www.cnblogs.com/rwy233/p/6689641.html
