Ural 1018. A Binary Apple Tree

这道题,可以马上看出是树形Dp,f[i,j]表示以i为根的子树(还包括 [i与 i的父亲] 这条边)内,保存j条边最多可以有多少苹果,显然f[i,j]=max(f[left[i],k]+f[right[i],j-k-1])+apple[i];输出的结果是f[1][q+1],q+1是因为根节点还多了一个虚拟的边。还有，因为输入数据没告诉谁是父节点谁是子节点，要建成无向图。

#include <iostream>#include <cstdio>#include <cstring>#define maxn 102using namespace std;int tree[maxn][maxn],f[maxn][maxn],deg[maxn],n,q;void init(){ memset(tree,0,sizeof(tree)); memset(f,0,sizeof(f)); memset(deg,0,sizeof(deg));}void addedge(int a,int b,int c){ tree[a][b]=c; tree[b][a]=c; deg[a]++; deg[b]++;}void dfs(int s,int fa){ if(deg[s]==1) { for(int i=1; i<=q; ++i) f[s][i]=tree[fa][s]; return; } int l=0,r=0; for(int i=1; i<=n; ++i) { if(tree[s][i]&&i!=fa) { if(!l) l=i; else r=i; } } dfs(l,s); dfs(r,s); for(int i=1; i<=q; ++i) { for(int k=0; k<i; ++k) { if(f[l][k]+f[r][i-1-k]>f[s][i]) f[s][i]=f[l][k]+f[r][i-1-k]; } f[s][i]+=tree[fa][s]; }}int main(void){ int a,b,c; while(scanf("%d%d",&n,&q)!=EOF) { init(); q++; for(int i=0; i<n-1; ++i) { scanf("%d%d%d",&a,&b,&c); addedge(a,b,c); } dfs(1,0);// for(int i=0;i<=n;++i)// {// for(int j=0;j<=n;++j)// {// cout<<f[j][i]<<" ";// }// cout<<endl;// } printf("%d/n",f[1][q]); } return 0;}