URAL_1018 BinaryAppleTree ( 树形dp )

vj or URAL

题意

给定一个以1为根节点共有 n 个结点的树, 去掉一些边, 最后剩下 m 条边, 使得所剩边权和最大

题解

按照题目要求建标准二叉树, 用 vector 存图, 保存左右两个子节点

根据题目问题可以设计基本状态 dp[i][j]，表示在第i节点下包含j条树枝最多可有多少个苹果。

根据状态易得状态转移方程：

                      |  0  ,              j = 0 || i为叶子节点
     dp[i][j] =       |                     仅保留左子树或右子树
                      |                     由左右子树共同组成      

代码

#include <bits/stdc++.h> 
using namespace std;
#define rg register 
#define sc scanf 
#define pf printf 
typedef long long ll;

const int maxn = 111;
int dp[maxn][maxn] = { 0 }, n, q;

struct NODE {
	int id, num;
	NODE ( int i, int n ) :id ( i ), num ( n ) {}
};
vector<NODE> s[maxn];

void dfs ( int x, int fa ) {
	int len = s[x].size ( );
	int c = 0;
	for ( int i = 0; i < len; i++ ) {
		int v = s[x][i].id;
		if ( v == fa ) {
			c = s[x][i].num;
			break;
		}
	}
	dp[x][1] = c;
	for ( int i = 0; i < len; i++ ) {
		int v = s[x][i].id;
		if ( v == fa )
			continue;
		dfs ( v, x );
		for ( int j = n; j >= 2; j-- ) {
			for ( int k = 1; k < j; k++ ) {
				dp[x][j] = max ( dp[x][j], dp[x][k] + dp[v][j - k] );
			}
		}
	}
}

int main ( ) {
	int a, b, c;
	sc ( "%d %d", &n, &q );

	for ( int i = 1; i < n; i++ ) {
		sc ( "%d %d %d", &a, &b, &c );
		s[a].push_back( NODE( b, c ) );
		s[b].push_back( NODE( a, c ) );
	}

	dfs ( 1, 0 );

	pf( "%d\n", dp[1][q + 1] );

	return 0;
}