贪吃的九头龙

贪吃的九头龙

---

题目描述

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

图一描述了果树的形态，图二描述了最优策略。

---

输入格式

第 $1$ 行包含三个整数 $N$ ，$M$ ，$K$ 。
$N$ 个果子依次编号 $1$~$N$ ，且最大的果子的编号总是 $1$。
第 $2$ 行到第 $N$ 行描述了果树的形态，每行包含三个整数 $a$ ，$b$ ，$c$ ，表示存在一段难受值为 $c$ 的树枝连接果子 $a$ 和果子$b$ 。

---

输出格式

一行，包含一个整数，表示在满足“大头”的要求的前提下，九头龙的难受值的最小值。
如果无法满足要求，输出 -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

---

样例输出

4

---

数据范围

$1\leq N\leq 300$; $2\leq M\leq N$ ; $1\leq K\leq N$ ; $0\leq c\leq 10^5$

---

Solution

首先，判断无解的条件：$n-k<m-1$。
状态表示：
用 $f[i][j][k]$ 表示当前在节点 $i$，而大头还需要吃 $j$ 个果子， k <script type="math/tex" id="MathJax-Element-1115">k</script> 表示当前节点的父节点是否被大头吃。
状态转移方程就很简单了。

---

Code

#include <iostream>
#include <cstdio>
#include <cstring>

#define Max(x,y) ((x)>(y)?(x):(y))
#define Min(x,y) ((x)<(y)?(x):(y))

using namespace std;

int n,m,k,cnt,cnts,f[900][900][4],fa[900];
int nxt[900],data[900],flow[900],head[900];
int nxts[900],datas[900],flows[900],heads[900];
int child[900][900],ans;

void add(int x,int y,int z){
    nxt[cnt]=head[x];data[cnt]=y;flow[cnt]=z;head[x]=cnt++;
}

void add1(int x,int y,int z){
    nxts[cnts]=heads[x];datas[cnts]=y;flows[cnts]=z;heads[x]=cnts++;
    nxts[cnts]=heads[y];datas[cnts]=x;flows[cnts]=z;heads[y]=cnts++;
}

void dfs(int now,int pre){
    int first=-1;
    for(int i=heads[now];i!=-1;i=nxts[i])if(datas[i]!=pre){
        child[now][datas[i]]=flows[i];
        fa[datas[i]]=now;
        if(first==-1){
            add(now,datas[i],flows[i]);
            first=datas[i];
        }
        else{
            add(first,datas[i],flows[i]);
            first=datas[i];
        }
        dfs(datas[i],now);
    }
}

int dp(int now,int need,bool turn){
    if(f[now][need][turn]!=-1)return f[now][need][turn];
    int tmp1=0,tmp2=0,p=-1,br=-1;
    if(turn==1)tmp1=child[fa[now]][now];
    else if(m==2)tmp2=child[fa[now]][now];
    for(int j=head[now];j!=-1;j=nxt[j])if(child[now][data[j]]!=-1){
        p=data[j];
        break;
    }
    for(int j=head[now];j!=-1;j=nxt[j])if(child[fa[now]][data[j]]!=-1){
        br=data[j];
        break;
    }
    if(need==0){
        if(p==-1&&br!=-1){
            tmp2+=dp(br,need,turn);
        }
        else if(p!=-1&&br==-1){
            tmp2+=dp(p,need,0);
        }
        else if(p!=-1&&br!=-1){
            int zz=0x3f3f3f3f;
            zz=Min(zz,dp(br,0,turn)+dp(p,0,0));tmp2+=zz;
        }
        else return f[now][need][turn]=Min(0x3f3f3f3f,tmp2);
        return f[now][need][turn]=Min(0x3f3f3f3f,tmp2);
    }
    if(p==-1&&br!=-1){
        tmp1+=dp(br,need-1,turn);
        tmp2+=dp(br,need,turn);
    }
    else if(p!=-1&&br==-1){
        tmp1+=dp(p,need-1,1);
        tmp2+=dp(p,need,0);
    }
    else if(p!=-1&&br!=-1){
        int zz=0x3f3f3f3f;
        for(int i=0;i<need;i++)zz=Min(zz,dp(br,need-i-1,turn)+dp(p,i,1));tmp1+=zz;
        zz=0x3f3f3f3f;
        for(int i=0;i<=need;i++)zz=Min(zz,dp(br,need-i,turn)+dp(p,i,0));tmp2+=zz;
    }
    else{
        if(need==1)return f[now][need][turn]=Min(0x3f3f3f3f,tmp1);
        else if(!need)return f[now][need][turn]=Min(0x3f3f3f3f,tmp2);
        else return f[now][need][turn]=0x3f3f3f3f;
    }
    return f[now][need][turn]=Min(0x3f3f3f3f,Min(tmp2,tmp1));
}

int main(){
    freopen("dragon.in","r",stdin);
    freopen("dragon.out","w",stdout);
    memset(heads,-1,sizeof heads);
    memset(head,-1,sizeof head);
    memset(f,-1,sizeof f);
    memset(child,-1,sizeof child);
    scanf("%d%d%d",&n,&m,&k);
    if(n-k<m-1){printf("-1");return 0;}
    for(int i=1,x,y,z;i<n;i++){
        scanf("%d%d%d",&x,&y,&z);
        add1(x,y,z);
    }
    dfs(1,0);
    for(int i=head[1];i!=-1;i=nxt[i])if(child[1][data[i]]!=-1){
        ans+=dp(data[i],k-1,1);
        break;
    }
    printf("%d\n",ans);
    return 0;
}