[树形dp] Jzoj P1162 贪吃的九头龙
题解
- 把题目简化一下:
- 将一棵树的结点染成m种颜色,每个结点只有一种颜色,在一条边两边的结点的颜色相同会产生费用
- (1)第1个结点必须是1颜色
- (2)必须有k个1颜色
- (3)每种颜色必须有一个结点
- 那么,可以发现一个性质
- 如果m>2,那么对答案有贡献的只有和1相连的边
- 其他的边不会产生其他的价值
- 如果m=2,那么只有1,2两种颜色
- 设f[i][j][0/1]为以i为根的子树中,有j个1颜色,根染颜色1或不染颜色1的价值
- 就很容易推出式子
- 与同颜色相连的,要加上边的价值;如果不是,就不加
代码
1 #include<cstdio> 2 #include<iostream> 3 #include<algorithm> 4 #include<cmath> 5 #include<cstring> 6 using namespace std; 7 struct edge { int to,from,v; }e[610]; 8 int n,m,k,size[310],f[310][310][2],head[310],cnt; 9 void insert(int x,int y,int z) { e[++cnt].to=y; e[cnt].from=head[x]; e[cnt].v=z; head[x]=cnt; } 10 void dp(int x,int fa) 11 { 12 int k[310][2]; 13 size[x]=1; f[x][1][1]=f[x][0][0]=0; 14 for (int i=head[x];i;i=e[i].from) 15 { 16 int son=e[i].to; 17 if (son==fa) continue; 18 dp(son,x); 19 size[x]+=size[son]; 20 memcpy(k,f[x],sizeof(k)); 21 memset(f[x],60,sizeof(f[x])); 22 int p=0; if (m==2) p=e[i].v; 23 for (int j=size[x];j>=0;j--) 24 { 25 if (j>0) for (int w=j-1;w>=0;w--) f[x][j][1]=min(f[x][j][1],k[j-w][1]+min(f[son][w][1]+e[i].v,f[son][w][0])); 26 for (int w=j;w>=0;w--) f[x][j][0]=min(f[x][j][0],k[j-w][0]+min(f[son][w][0]+p,f[son][w][1])); 27 } 28 } 29 } 30 int main() 31 { 32 scanf("%d%d%d",&n,&m,&k); 33 if (n-k<m-1) { printf("-1\n"); return 0; } 34 for (int i=1;i<=n-1;i++) 35 { 36 int x,y,z; 37 scanf("%d%d%d",&x,&y,&z); 38 insert(x,y,z); insert(y,x,z); 39 } 40 memset(f,60,sizeof(f)); 41 dp(1,0); 42 printf("%d",f[1][k][1]); 43 return 0; 44 }