当前位置: 代码迷 >> 综合 >> 树形DP依赖背包 洛谷 P2015 二叉苹果树
  详细解决方案

树形DP依赖背包 洛谷 P2015 二叉苹果树

热度:43   发布时间:2024-01-15 07:28:54.0

题目描述

有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点)

这棵树共有N个结点(叶子点或者树枝分叉点),编号为1-N,树根编号一定是1

我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树

2   5

 \ /

  3   4

   \ /

    1

现在这颗树枝条太多了,需要剪枝。但是一些树枝上长有苹果。

给定需要保留的树枝数量,求出最多能留住多少苹果。

输入输出格式

输入格式:
 

12个数,NQ(1<=Q<= N,1<N<=100)

N表示树的结点数,Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。

每行3个整数,前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。

每根树枝上的苹果不超过30000个。

输出格式:
 

一个数,最多能留住的苹果的数量。

输入输出样例

输入样例#1 复制

5 2

1 3 1

1 4 10

2 3 20

3 5 20

输出样例#1 复制

21

算法分析:

 树形依赖背包的入门题,参考解析:选课

但与上一题有所不同,

1.这题明确给出根节点为1,所以不会是上一题的无根树

2.父节点和子结点不像上一题一样标准的给出,所以每一个结点就没有了明确的值。

3.这题要求删边对应上一题删点数目+1

这题遇上题一模一样,但是这题目没给标准的给你父节点和子节点,但在后面给你一个价值,这时候你不确定子节点是谁,所以值不能想上一题一样直接复制子节点,所以vextor就不好入手了,所以建树时要注意。

这就需要完整的链式向前星,需要保存边的值,在建好树后,赋值给对应的子节点。

推荐代码实现:

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
const int N=310;
struct node
{int v;///终端点int next;///下一条同样起点的边号int w;///权值
}edge[N*2];///无向边,2倍
int head[N];///head[u]=i表示以u为起点的所有边中的第一条边是 i号边
int tot;  ///总边数
int minn;
void add(int u,int v,int w)
{edge[tot].v=v;edge[tot].w=w;edge[tot].next=head[u];head[u]=tot++;
}
int n,m;
int dp[N][N],val[N];
void dfs(int u,int fa) 
{//dp[u][1]=val[u];///选一个肯定选自己这个结点for(int i=head[u];i!=-1;i=edge[i].next){int v=edge[i].v;if(fa==v) continue;   ///如果下一个相邻节点就是父节点,则证明到底层了,开始递归父节点的兄弟节点dp[v][1]=edge[i].w;    ///每一个结点的赋值dfs(v,u);for(int j=m;j>0;j--)    ///背包容量///下面这个循环必须是k<j结束,不能取等号,因为至少需要留一个点来取u点for(int k=0;k<j;k++)   ///选择用户{ dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]);}}}
int main()
{while(~scanf("%d%d",&n,&m)){memset(head,-1,sizeof(head));memset(dp,0,sizeof(dp)); tot=0;for(int i=1;i<n;i++){int u,w,v;scanf("%d%d%d",&u,&v,&w);add(u,v,w);add(v,u,w);///注意题目给出第一个不一定是父亲结点	}m+=1;   ///虚结点加1,删n条边相当于删除n个点dfs(1,-1);printf("%d\n",dp[1][m]);}return 0;
}

我一开始做的代码

代码实现:

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
const int N=110;
struct node
{int v;///终端点int next;///下一条同样起点的边号int w;///权值
}edge[N*2];///无向边,2倍
int head[N];///head[u]=i表示以u为起点的所有边中的第一条边是 i号边
int tot;  ///总边数
void add(int u,int v,int w)
{edge[tot].v=v;edge[tot].w=w;edge[tot].next=head[u];head[u]=tot++;
}
int n,m;
int dp[N][N];
///dp[i][j]表示节点i保留j个枝条的所剩苹果最大值
int dfs(int u,int fa) 
{int num=0;  ///num表示u节点的子节点数目for(int i=head[u];i!=-1;i=edge[i].next){int v= edge[i].v;if(fa==v) continue;   ///如果下一个相邻节点就是父节点,则证明到底层了,开始递归父节点的兄弟节点num+=dfs(v,u)+1;for(int j=min(num,m);j>=1;j--)  ///注意删除有限制for(int k=min(j-1,num);k>=0;k--){dp[u][j]=max(dp[u][j],dp[u][j-k-1]+dp[v][k]+edge[i].w);}}return num;
}int main()
{scanf("%d%d",&n,&m);memset(head,-1,sizeof(head));memset(dp,0,sizeof(dp));tot=0;for(int i=1;i<n;i++){int u,v,w;scanf("%d%d%d",&u,&v,&w);add(u,v,w);add(v,u,w);}dfs(1,-1);printf("%d\n",dp[1][m]);return 0;
}