Let's imagine how apple tree looks in binary computer world. You're right, it looks just like a binary tree, i.e. any biparous branch splits up to exactly two new branches. We will enumerate by integers the root of binary apple tree, points of branching and the ends of twigs. This way we may distinguish different branches by their ending points. We will assume that root of tree always is numbered by 1 and all numbers used for enumerating are numbered in range from 1 to N, whereN is the total number of all enumerated points. For instance in the picture belowN is equal to 5. Here is an example of an enumerated tree with four branches:
2 5
\ /
3 4
\ /
1
|
题目大意:题目中给定了一棵根节点为1的标准二叉树(所有非叶子节点都有两个儿子)。每一条边有一定的边权。问保留Q条树枝的情况下最大的权值为多少?
解题思路:按照题目要求建树,因为是标准二叉树,所以我们可以开两个数组记录每个节点的左儿子和右儿子。根据题目问题可以设计基本状态 dp[i][j],表示在第i节点下包含j条树枝最多可有多少个苹果。
#include <bits/stdc++.h>
#define Fori(x) for(int i=0;i<x;i++)
#define Forj(x) for(int j=0;j<x;j++)
#define maxn 105
#define inf 0x3f3f3f3f
#define ONES(x) __builtin_popcount(x)
using namespace std;
typedef long long ll ;
const double eps =1e-8;
const int mod = 1000000007;
typedef pair<int, int> P;
const double PI = acos(-1.0);
int dx[4] = {0,0,1,-1};
int dy[4] = {1,-1,0,0};
int n,q;
int ans;
int a[maxn][maxn];//存储边权
int dp[maxn][maxn];//dp[i][j] -> 在第i节点下包含j条树枝最多可有多少个苹果
int l[maxn],r[maxn];//存储每个节点的左右子节点
int x,y,z;
vector<int> e[maxn];
void dfs(int u, int fa)// build a tree
{
for(int i = 0; i<e[u].size(); i++)
{
int v = e[u][i];
if(v==fa) continue;
if(l[u]==0)
l[u] = v;
else
r[u] = v;
dfs(v, u);
}
}
int cal(int u, int num)
{
if(num==0 || l[u]+r[u]==0) return 0;
if(dp[u][num]) return dp[u][num];
for(int i = 0; i<=num-2; i++)
{
dp[u][num] = max(dp[u][num], a[u][l[u]] + a[u][r[u]] + cal(l[u], i) + cal(r[u], num -2 - i));
}
dp[u][num] = max(dp[u][num], a[u][l[u]] + cal(l[u], num-1) );
dp[u][num] = max(dp[u][num], a[u][r[u]] + cal(r[u], num-1) );
return dp[u][num];
}
int main()
{
//freopen("test.txt","r",stdin);
cin>>n>>q;
for(int i = 1; i<n; i++)
{
cin>>x>>y>>z;
a[x][y] = a[y][x] = z;
e[x].push_back(y);
e[y].push_back(x);
}
dfs(1,-1);
cout << cal(1,q) << endl;
return 0;
}