POJ2486 Apple Tree 【树上背包】
一句话题意:一棵树,一共n个点,每个点上有一个权值,求从1出发,走k步,最多能遍历到的权值。可以往回走。
第一(二)道树上背包题,先是看了dalao的题解,改了一点就过样例了。然而....TLE??? 改了挺久发现由于多组数据且没有“0 0”的输入,如果不在读入的时候加“~”或“EOF”就会死循环,从而导致TLE。
状态设计:设f[i][j][0/1]为以i为根的子树上,走j步,能得到的最大权值(0/1的表示会在转移方程中描述) 考虑:(此处参考dalao@zubizakeli ,侵删qwq)每个节点在最终答案中的类型:1,不经过;2,经过但不返回;3,经过且返回 (返回的定义是最终的停止节点不位于该节点的子树中) 那么可以进行转移:
之后便是一些细节问题:给数组赋初值(从0开始!走0步),更新head数组。
实现还是比较简单的啦。
code
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
1 #include<cstdio> 2 #include<algorithm> 3 #include<cstring> 4 5 using namespace std; 6 7 int n,k,tot; 8 int w[300],head[300]; 9 int f[300][200][3]; 10 struct node{ 11 int to,next,val; 12 }edge[300]; 13 14 void add(int x,int y) 15 { 16 edge[++tot].to=y; 17 edge[tot].next=head[x]; 18 head[x]=tot; 19 } 20 21 void read(int &x) 22 { 23 x=0; 24 char ch=getchar(); 25 bool flag=false; 26 while(ch<'0'||ch>'9') flag|=(ch=='-'),ch=getchar(); 27 while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch^48),ch=getchar(); 28 x=flag ? -x : x; 29 } 30 31 void TreeDp(int u,int fa) 32 { 33 for(int i=0;i<=k;i++) f[u][i][1]=w[u],f[u][i][0]=w[u]; 34 for(int i=head[u];i;i=edge[i].next) 35 { 36 int v=edge[i].to; 37 if(v==fa) continue; 38 TreeDp(v,u); 39 for(int kk=k;kk>=0;kk--) 40 for(int j=0;j<=kk;j++) 41 { 42 if(kk>=j+2) f[u][kk][0]=max(f[u][kk][0],f[v][j][0]+f[u][kk-j-2][0]); 43 if(kk>=j+2) f[u][kk][1]=max(f[u][kk][1],f[v][j][0]+f[u][kk-j-2][1]); 44 if(kk>=j+1) f[u][kk][1]=max(f[u][kk][1],f[v][j][1]+f[u][kk-j-1][0]); 45 } 46 } 47 } 48 49 void init() 50 { 51 memset(f,0,sizeof(f)); 52 memset(head,0,sizeof(head)); 53 tot=0; 54 } 55 56 int main() 57 { 58 while(scanf("%d%d",&n,&k)!=EOF) 59 { 60 for(int i=1;i<=n;i++) read(w[i]); 61 for(int i=1;i<=n-1;i++) 62 { 63 int x=0,y=0; 64 read(x),read(y); 65 add(x,y),add(y,x); 66 } 67 TreeDp(1,-1); 68 printf("%d\n",max(f[1][k][0],f[1][k][1])); 69 init(); 70 } 71 return 0; 72 }
小结:分类讨论常常也是解题重要的突破口呐。
独立意志与自由思想是必须争的,且须以生死力争。