码迷,mamicode.com
首页 > 移动开发 > 详细

二叉苹果树(codevs 5565)树形DP

时间:2018-08-09 23:17:20      阅读:206      评论:0      收藏:0      [点我收藏+]

标签:main   oid   out   留住   不能   父节点   color   code   mes   

题目描述 Description

      有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点)这棵树共有N个结点(叶子点或者树枝分叉点),编号为1-N,树根编号一定是1。我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。现在这颗树枝条太多了,需要剪枝。但是一些树枝上长有苹果。
       给定需要保留的树枝数量,求出最多能留住多少苹果。

 

输入描述 Input Description

第1行2个数,N和Q(1<=Q<= N,1<N<=100)。
N表示树的结点数,Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数,前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。

 

输出描述 Output Description

剩余苹果的最大数量。

 

样例输入 Sample Input

5 2

1 3 1

1 4 10

2 3 20

3 5 20

 

样例输出 Sample Output

21

 

数据范围及提示 Data Size & Hint

对于20%数据 n<=20;

对于100%数据1<N<=100,1<=Q<= N.


所遇到的问题:给出的节点没有父子关系,需要建双向边走dfs的时候判断,这部分由其注意,要先判断是否回父,在更新父亲节点
 

代码如下

#include<stdio.h>
#include<algorithm>
using namespace std;

int n,q,cnt,f[210][210],fa[210],first[210];
struct Edge{
    int to,next,val;
}edge[210];

void add(int from,int to,int val)
{
    edge[++cnt].to=to;
    edge[cnt].val=val;
    edge[cnt].next=first[from];
    first[from]=cnt;
}

void dfs(int from)
{
    for(int i=first[from];i;i=edge[i].next) {
        int to=edge[i].to;
        if(to == fa[from]) continue;
        fa[to]=from;
        dfs(to);
        for(int j=q;j>=1;--j)//采用倒推,j为总边数
            for(int k=0;k<j;++k)//k为子节点支数,不能超过总支数 
                f[from][j]=max(f[from][j],f[to][k]+f[from][j-k-1]+edge[i].val);//f[to][k] 子节点拥有 k 支时的 val, f[from][j-k-1] 父节点剩余支的 val,edge[i].val当前支的val 
    }
}


int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<n;++i) {
        int from,to,val;
        scanf("%d%d%d",&from,&to,&val);
        add(from,to,val);
        add(to,from,val);
    }
    dfs(1);
    printf("%d",f[1][q]);
    return 0;
}

 

二叉苹果树(codevs 5565)树形DP

标签:main   oid   out   留住   不能   父节点   color   code   mes   

原文地址:https://www.cnblogs.com/qseer/p/9451768.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!