标签：
题目链接：Codeforces Round #369 (Div. 2) C. Coloring Trees
题意：
有n个树，每个树有一个颜色，如果颜色值为0，表示没有颜色，一共有m个颜色，第j种颜色涂第i棵树需要花费pij，颜色一样且相邻的分为一组
现在要将所有颜色为0的树涂上颜色，使得这些树恰好可以分为k组，问最小的花费
题解：
考虑dp[i][j][k]，表示考虑第i棵树涂第j种颜色，当前分为k组的最小花费，然后状态转移看代码，注意的是dp的初始状态
1 #include<bits/stdc++.h> 2 #define F(i,a,b) for(int i=a;i<=b;++i) 3 using namespace std; 4 typedef long long ll; 5 6 int n,m,d; 7 ll dp[111][111][111],v[101][101],inf=1e18; 8 int dt[101]; 9 10 inline void up(ll &a,ll b){if(a>b)a=b;} 11 12 int main(){ 13 scanf("%d%d%d",&n,&m,&d); 14 F(i,1,n)scanf("%d",dt+i); 15 F(i,1,n)F(j,1,m)scanf("%lld",&v[i][j]); 16 F(i,0,n)F(j,0,m)F(k,0,d)dp[i][j][k]=inf; 17 if(dt[1])dp[1][dt[1]][1]=0; 18 else F(i,1,m)dp[1][i][1]=v[1][i]; 19 F(i,2,n)F(j,1,m)F(k,1,d)if(dp[i-1][j][k]<inf) 20 { 21 if(dt[i])up(dp[i][dt[i]][k+(dt[i]!=j)],dp[i-1][j][k]); 22 else F(ii,1,m)up(dp[i][ii][k+(ii!=j)],dp[i-1][j][k]+v[i][ii]); 23 } 24 ll ans=inf; 25 F(i,1,m)up(ans,dp[n][i][d]); 26 if(ans==inf)puts("-1"); 27 else printf("%lld\n",ans); 28 return 0; 29 }
Codeforces Round #369 (Div. 2) C. Coloring Trees(dp)
标签：
原文地址：http://www.cnblogs.com/bin-gege/p/5822645.html
