CodeVs 1746 贪吃的九头龙 （树型dp 多叉转二叉）

1746 贪吃的九头龙 2002年NOI全国竞赛

时间限制: 2 s
空间限制: 128000 KB

题目描述 Description
传说中的九头龙是一种特别贪吃的动物。虽然名字叫“九头龙”，但这只是说它出生的时候有九个头，而在成长的过程中，它有时会长出很多的新头，头的总数会远大于九，当然也会有旧头因衰老而自己脱落。
有一天，有M 个脑袋的九头龙看到一棵长有N 个果子的果树，喜出望外，恨不得一口把它全部吃掉。可是必须照顾到每个头，因此它需要把N 个果子分成M组，每组至少有一个果子，让每个头吃一组。
这M个脑袋中有一个最大，称为“大头”，是众头之首，它要吃掉恰好K个果子，而且K个果子中理所当然地应该包括唯一的一个最大的果子。果子由N-1根树枝连接起来，由于果树是一个整体，因此可以从任意一个果子出发沿着树枝“走到”任何一个其他的果子。
对于每段树枝，如果它所连接的两个果子需要由不同的头来吃掉，那么两个头会共同把树枝弄断而把果子分开；如果这两个果子是由同一个头来吃掉，那么这个头会懒得把它弄断而直接把果子连同树枝一起吃掉。当然，吃树枝并不是很舒服的，因此每段树枝都有一个吃下去的“难受值”，而九头龙的难受值就是所有头吃掉的树枝的“难受值”之和。
九头龙希望它的“难受值”尽量小，你能帮它算算吗？
例如图 1 所示的例子中，果树包含8 个果子，7 段树枝，各段树枝的“难受值”标记在了树枝的旁边。九头龙有两个脑袋，大头需要吃掉4个果子，其中必须包含最大的果子。即N=8，M=2，K=4：
图一描述了果树的形态，图二描述了最优策略。
大头吃4个果子，用实心点标识；
小头吃4个果子，用空心点标识；
九头龙的难受值为4，因为图中用细边标
记的树枝被大头吃掉了。

输入描述 Input Description
输入文件dragon.in的第1行包含三个整数N (1<=N<=300)，M (2<=M<=N)，K (1<=K<=N)。 N 个果子依次编号1,2,…,N，且最大的果子的编号总是1。第2行到第N行描述了果树的形态，每行包含三个整数a (1<=a<=N)，b (1<=b<=N)，c (0<=c<=105)，表示存在一段难受值为c的树枝连接果子a和果子b。

输出描述 Output Description
输出文件dragon.out 仅有一行，包含一个整数，表示在满足“大头”的要求的前提下，九头龙的难受值的最小值。如果无法满足要求，输出-1。

样例输入 Sample Input
8 2 4
1 2 20
1 3 4
1 4 13
2 5 10
2 6 12
3 7 15
3 8 5

样例输出 Sample Output
4

数据范围及提示 Data Size & Hint
该样例对应于题目描述中的例子。

思路：
[ oyyl的链接 ]
oyyl的链接

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 310
using namespace std;

int n , m , k;
int dp[M][M][2] , inf = 100000000;
int head[M], idc;

struct node{
    int  ls, rs, siz, w;
}nd[M];

struct Edge{
    int to, nxt, w;
}ed[M << 1];

void add(int u, int v, int w){
    ed[++idc].to = v;
    ed[idc].nxt = head[u];
    ed[idc].w = w;
    head[u] = idc;
}

void rebuild(int u, int f){
    nd[u].siz = 1;
    for(int i = head[u]; i; i = ed[i].nxt){
        int v = ed[i].to;
        if(v == f) continue;
        rebuild(v, u);
        nd[u].siz += nd[v].siz;
        nd[v].w = ed[i].w;
        nd[v].rs = nd[u].ls;//转二叉 
        nd[u].ls = v;
    }
}

int dfs_dp(int v, int num, int g){//dp[i][j][0/1] 代表在第i号节点【被吃1/没被吃0】，大头在这颗子树中吃了j个果子；
    if(num < 0) return inf;
    if(dp[v][num][g] >= 0) return dp[v][num][g];//记忆化 
    if(v == 0 && num == 0) return dp[v][num][g] = 0;
    if(v == 0 && num != 0) return dp[v][num][g] = inf;//buhefa 
    dp[v][num][g] = inf;
    for(int i=0; i<=min(nd[v].siz, num); ++i){//枚举在当前子树中分配多少黑点 
        int t1 = dfs_dp(nd[v].ls, i-1, 1) + (g == 1) * nd[v].w;//若当前点和父节点都为黑，就要加上边权 
        int t2 = dfs_dp(nd[v].ls, i, 0) + (m == 2) * (g == 0) * nd[v].w;//若m==2当前点和父节点都不为黑，也要加上边权 
        int t3 = dfs_dp(nd[v].rs, num-i, g);//到兄弟节点的状态 
        t1 = min(t1, t2);
        dp[v][num][g] = min(dp[v][num][g] , t1 + t3);
    }
    return dp[v][num][g];
}

int main(){
    memset(dp, -100, sizeof dp);
    scanf("%d%d%d", &n, &m, &k);
    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);
    }
    if(n-k < m-1) { printf("-1\n"); return 0;}
    rebuild(1 , 0);
    printf("%d",dfs_dp(nd[1].ls, k-1, 1));
}