题目:
有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点)
这棵树共有N个结点(叶子点或者树枝分叉点),编号为1-N,树根编号一定是1。
我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树
2 5
\ /
3 4
\ /
1
现在这颗树枝条太多了,需要剪枝。但是一些树枝上长有苹果。
给定需要保留的树枝数量,求出最多能留住多少苹果。
输入格式:
第1行2个数,N和Q(1<=Q<= N,1<N<=100)。
N表示树的结点数,Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数,前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。
输出格式:
一个数,最多能留住的苹果的数量。
输入样例:
5 2
1 3 1
1 4 10
2 3 20
3 5 20
输出样例:
21
分析:
dp+背包
注意:
next数组开到2*maxn!!!
注意思考动态转移方程
#include<bits/stdc++.h>
#define maxn 110
using namespace std;
struct node{
int u,v,w;
};
node edge[maxn*2];
int n,m;
int head[maxn],next[maxn*2],dp[maxn][maxn],s[maxn];
void work(int k){
int x=0,z,x1=0,z1,tot=1;
for(int i=head[k];i!=0;i=next[i]){
int g=edge[i].v;
if(s[g]==0){
s[g]=1;work(g);
if(tot==1){x=g;z=edge[i].w;}
if(tot==2){x1=g;z1=edge[i].w;}
tot++;
}
}
if(x!=0)dp[k][1]=max(dp[x][0]+z,dp[x1][0]+z1);
for(int i=2;i<=m;i++)
if(x!=0)
dp[k][i]=max(max(dp[x][i-1]+z,dp[x1][i-1]+z1),dp[x][i-2]+z+dp[x1][i-2]+z1);
return ;
}
int main(){
scanf("%d %d",&n,&m);
int tot=0;
for(int i=1;i<=n;i++){
int x,y,z;
scanf("%d %d %d",&x,&y,&z);
edge[++tot].u=x;edge[tot].v=y;edge[tot].w=z;
next[tot]=head[x];head[x]=tot;
edge[++tot].u=y;edge[tot].v=x;edge[tot].w=z;
next[tot]=head[y];head[y]=tot;
}
s[1]=1;work(1);
printf("%d\n",dp[1][m]);
return 0;
}