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

最新推荐文章于 2022-11-22 23:15:14 发布

Rutifrl

最新推荐文章于 2022-11-22 23:15:14 发布

阅读量232

点赞数

分类专栏： ACM 文章标签： ACM dp

本文链接：https://blog.csdn.net/Rutifrl/article/details/52437457

版权

ACM 专栏收录该内容

60 篇文章 0 订阅

订阅专栏

/*
dp[i][j][k]表示当前处理第i棵树，第i棵树是第j种颜色的代价
如果当前值非0,不用处理花费dp[i][c[i][k]=min(dp[i][c[i][k],dp[i-1][j][k-1])
如果当前值是0,要处理花费dp[i][now][k]=min(dp[i][now][k],dp[i-1][last][k]+p[i][now]),
dp[i][now][k]=min(dp[i][now][k],dp[i-1][last][k-1]+p[i][now])
*/
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<iostream>
using namespace std;
#define LL long long
LL inf=1e18;
int n,m,K;
LL co[105];
LL dp[105][105][105];
LL p[105][105];

int main()
{
    LL i,j,k,c;
    cin>>n>>m>>K;
    for(i=1;i<=n;i++)
        cin>>co[i];
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=m;j++)
            scanf("%d",&p[i][j]);
    }
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++)
            for(k=0;k<=n;k++)
                dp[i][j][k]=inf;
    for(i=1;i<=n;i++)
    {
        for(k=1;k<=i;k++)
        {
            if(co[i])
            {
                dp[i][co[i]][k]=dp[i-1][co[i]][k];
                for(j=1;j<=m;j++)
                    if(co[i]!=j)
                        dp[i][co[i]][k]=min(dp[i][co[i]][k],dp[i-1][j][k-1]);
            }
            else
            {
                for(j=1;j<=m;j++)
                    for(c=1;c<=m;c++)
                        if(j==c) dp[i][c][k]=min(dp[i][c][k],dp[i-1][j][k]+p[i][c]);
                        else dp[i][c][k]=min(dp[i][c][k],dp[i-1][j][k-1]+p[i][c]);
            }
        }
    }

    LL ans=inf;
    for(i=1;i<=m;i++)
        ans=min(ans,dp[n][i][K]);
    if(ans>=inf)
        ans=-1;
    cout<<ans;
}