【树形dp】贪吃的九头龙

描述

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

输入格式

输入的第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。
输出格式

输出仅有一行，包含一个整数，表示在满足“大头”的要求的前提下，九头龙的难受值的最小值。如果无法满足要求，输出-1。
样例1

样例输入1

8 2 4
1 2 20
1 3 4
1 4 13
2 5 10
2 6 12
3 7 15
3 8 5
Copy
样例输出1

4
Copy
提示

树形动态规划

分析：
首先，当小头大于一个时，都让小头来吃，肯定不会有难受值。因为小头可以隔一个分一个。（当时为什么没有分析出来呢？）。这样我们就把问题转化成了
m==2时 大头+小头
m>2时 只需要考虑大头吃k个
无解，不够吃的：K+M-1>N;
f[i][j][k]:表示以i为根的子树选出j个果子（包括自身）时最小的难受值（k=1表示他爸爸被大头吃，k=0表示他爸爸被小头吃）

int t1=dfs(node[root].lc,i,0)+(m==2)*(flag==0)*node[root].dis;
int t2=dfs(node[root].lc,i-1,1)+(flag==1)*node[root].dis;
int t3=dfs(node[root].rc,k-i,flag);
f[root][k][flag]=min(f[root][k][flag],min(t1,t2)+t3);

这些细节都要注意。 还有多叉树转二叉树的方法。
还是要看题解

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=301;
const int INF=100000000;//原来是INF开的太大了 
int n,m,k,x,y,z,f[MAXN][MAXN][2],sum[MAXN];
struct Tree{
    int lc,rc,dis;
}node[MAXN];

int dfs(int root,int k,bool flag)
{
    if (k<0) return INF;//这里return的是INF而不是0 
    if (f[root][k][flag]>=0) return f[root][k][flag];
    if (!root&&!k) return f[root][k][flag]=0;

    f[root][k][flag]=INF;
    for (int i=0; i<=min(k,sum[root]); i++)//可以取0，各种边界和细节 
    {
        int t1=dfs(node[root].lc,i,0)+(m==2)*(flag==0)*node[root].dis;
        int t2=dfs(node[root].lc,i-1,1)+(flag==1)*node[root].dis;
        int t3=dfs(node[root].rc,k-i,flag);
        f[root][k][flag]=min(f[root][k][flag],min(t1,t2)+t3);
    }
//  printf("%d\n",f[root][k][flag]);
    return f[root][k][flag];
}

void find_son(int root)//包括根节点本身 
{
    if (!root) return;
    find_son(node[root].lc);
    find_son(node[root].rc);
    sum[root]=sum[node[root].lc]+sum[node[root].rc]+1;
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    memset(f,255,sizeof(f));
//  f[0][0][0]=0; f[0][0][1]=0;

    for (int i=1; i<=n-1; i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        node[y].rc=node[x].lc;
        node[x].lc=y;
        node[y].dis=z;
    }
    if (k+m-1>n) {printf("-1\n"); return 0;}
    find_son(1);
//  for (int i=1; i<=n; i++)    printf("i:%d l:%d r:%d dis:%d\n",i,node[i].lc,node[i].rc,node[i].dis);
//  for (int i=1; i<=n; i++)    printf("%d ",sum[i]);
    printf("%d",dfs(node[1].lc,k-1,1));
    return 0;
}