贪吃的九头龙
题目描述
传说中的九头龙是一种特别贪吃的动物。
虽然名字叫“九头龙”,但这只是说它出生的时候有九个
头,而在成长的过程中,它有时会长出很多的新头,头的总数会远大于九,当然也会有旧头因衰老而自己脱落。
有一天,有
M
个脑袋的九头龙看到一棵长有
可是必须照顾到每个头,因此它需要把
N
个果子分成
这
M
个脑袋中有一个最大,称为“大头”,是众头之首,它要吃掉恰好
果子由
对于每段树枝,如果它所连接的两个果子需要由不同的头来吃掉,那么两个头会共同把树枝弄断而把果子分开;如果这两个果子是由同一个头来吃掉,那么这个头会懒得把它弄断而直接把果子连同树枝一起吃掉。
当然,吃树枝并不是很舒服的,因此每段树枝都有一个吃下去的“难受值”,而九头龙的难受值就是所有头吃掉的树枝的“难受值”之和。
九头龙希望它的“难受值”尽量小,你能帮它算算吗?
例如 图1 所示的例子中,果树包含
8
个果子,
即
图一描述了果树的形态,图二描述了最优策略。
输入格式
第
1
行包含三个整数
N
个果子依次编号
第
2
行到第
输出格式
一行,包含一个整数,表示在满足“大头”的要求的前提下,九头龙的难受值的最小值。
如果无法满足要求,输出 -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≤N≤300 ; 2≤M≤N ; 1≤K≤N ; 0≤c≤105
Solution
首先,判断无解的条件:
n−k<m−1
。
状态表示:
用
f[i][j][k]
表示当前在节点
i
,而大头还需要吃
状态转移方程就很简单了。
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;
}