Codeforces711 C. Coloring Trees（dp）

题意：

数据范围：n,m<=100，涂色花费颜料量a(i,j)<=1e9

解法：

令d[i][j][k]表示前i棵树,已经j段,末尾为颜色k的最小代价.
对于d[i][j][k],枚举第i+1棵树的颜色nt,
那么d[i][j][k]可以向d[i+1][j+(nt!=k)][nt]转移,转移代价为涂色代价.

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
int d[111][111][111];
int c[111][111];
int a[111];
int n,m,p;
signed main(){
    ios::sync_with_stdio(0);
    cin>>n>>m>>p;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            cin>>c[i][j];
        }
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<=p;j++){
            for(int k=0;k<=m;k++){
                d[i][j][k]=1e18;
            }
        }
    }
    d[0][0][0]=0;
    for(int i=0;i<n;i++){
        for(int j=0;j<=p;j++){
            for(int k=0;k<=m;k++){
                if(d[i][j][k]==1e18)continue;
                if(!a[i+1]){
                    for(int nt=1;nt<=m;nt++){
                        int f=(nt!=k);
                        if(j+f>p)continue;
                        d[i+1][j+f][nt]=min(d[i+1][j+f][nt],d[i][j][k]+c[i+1][nt]);
                    }
                }else{
                    int nt=a[i+1];
                    int f=(a[i+1]!=k);
                    if(j+f>p)continue;
                    d[i+1][j+f][nt]=min(d[i+1][j+f][nt],d[i][j][k]);
                }
            }
        }
    }
    int ans=1e18;
    for(int k=1;k<=m;k++){
        ans=min(ans,d[n][p][k]);
    }
    if(ans==1e18)ans=-1;
    cout<<ans<<endl;
    return 0;
}

---