codeforces C. Coloring Trees - 序列染色-指定颜色段的最小染料

有n棵树吗，m个染料。没有染色的树用0表示。用颜色i染第j棵树需要p[i][j]的染料。要求染成的颜色段为k。需要的最少染料。

思路：直接dp:dp[i][j][k]第i棵树染颜色j的前i棵一共有k个颜色段的最少染料。

#include <bits/stdc++.h>
#define LL long long
using namespace std;
 
int a[105];
int b[105][105];
LL f[105][105][105];
 
int main(){
 
    int n, m, K, L=0;
    scanf("%d%d%d", &n, &m, &K);
    for(int i=1; i<=n; i++){
        scanf("%d", &a[i]);
    }
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            scanf("%d", &b[i][j]);
        }
    }
 
    memset(f, -1, sizeof(f));
    if(a[1]==0){
        for(int i=1; i<=m; i++){
            f[1][i][1]=b[1][i];
        }
    }
    else{
        f[1][a[1]][1]=0;
    }
 
    for(int i=1; i<n; i++){
 
        if(a[i+1]){//已经染色
 
            for(int j=1; j<=m; j++){
                for(int k=1; k<=n; k++){
 
                    if(f[i][j][k]!=-1){//这个状态存在
 
                        if(a[i+1]==j){
                            if(f[i+1][a[i+1]][k]==-1){
                                f[i+1][a[i+1]][k]=f[i][j][k];
                            }
                            else{
                                f[i+1][a[i+1]][k]=min(f[i+1][a[i+1]][k], f[i][j][k]);
                            }
                        }
                        else{
 
                            if(f[i+1][a[i+1]][k+1]==-1){
                                f[i+1][a[i+1]][k+1]=f[i][j][k];
                            }
                            else{
                                f[i+1][a[i+1]][k+1]=min(f[i+1][a[i+1]][k+1], f[i][j][k]);
                            }
                        }
                    }
                }
            }
        }
        else{
 
            for(int j=1; j<=m; j++){
                for(int k=1; k<=n; k++){
                    if(f[i][j][k]!=-1){//这个状态存在
                        for(int w=1; w<=m; w++){
 
                            if(w==j){
                                if(f[i+1][w][k]==-1){
                                    f[i+1][w][k]=f[i][j][k]+b[i+1][w];
                                }
                                else{
                                    f[i+1][w][k]=min(f[i+1][w][k], f[i][j][k]+b[i+1][w]);
                                }
                            }
                            else{
 
                                if(f[i+1][w][k+1]==-1){
                                    f[i+1][w][k+1]=f[i][j][k]+b[i+1][w];
                                }
                                else{
                                    f[i+1][w][k+1]=min(f[i+1][w][k+1], f[i][j][k]+b[i+1][w]);
                                }
                            }
                        }
                    }
                }
            }
        }
    }
    LL ans=1ll<<60;
    for(int i=1; i<=m; i++){
        if(f[n][i][K]!=-1){
            ans=min(ans, f[n][i][K]);
        }
    }
    printf("%lld\n", ((ans==(1ll<<60))?-1:ans));
 
    return 0;
}