Codeforces Round #369 (Div. 2) C. Coloring Trees(dp)

题目链接：Codeforces Round #369 (Div. 2) C. Coloring Trees

题意：

有n个树，每个树有一个颜色，如果颜色值为0，表示没有颜色，一共有m个颜色，第j种颜色涂第i棵树需要花费p_ij，颜色一样且相邻的分为一组

现在要将所有颜色为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 }

View Code

 

转载于:https://www.cnblogs.com/bin-gege/p/5822645.html