CodeForces 711C Coloring Trees

简单$dp$。

$dp[i][j][k]$表示：前$i$个位置染完色，第$i$个位置染的是$j$这种颜色，前$i$个位置分成了$k$组的最小花费。总复杂度$O({n^4})$。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<iostream>
using namespace std;
typedef long long LL;
const double pi=acos(-1.0),eps=1e-;

LL INF=1e17;
LL dp[][][];
LL p[][];
int c[],n,m,k;

int main()
{
    scanf("%d%d%d",&n,&m,&k);

    for(int i=;i<=n;i++) scanf("%d",&c[i]);

    for(int i=;i<=n;i++)
        for(int j=;j<=m;j++) scanf("%lld",&p[i][j]);

    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            for(int s=;s<=k;s++)
            {
                dp[i][j][s]=INF;
            }
        }
    }

    if(c[]!=) dp[][c[]][]=;
    else
    {
        for(int j=;j<=m;j++) dp[][j][]=p[][j];
    }

    for(int i=;i<=n;i++)
    {
        if(c[i]==)
        {
            for(int j=;j<=m;j++)
            {
                for(int s=;s<=i;s++)
                {
                    LL tmp=INF;
                    tmp=min(tmp,dp[i-][j][s]);
                    for(int t=;t<=m;t++)
                    {
                        if(t==j) continue;
                        tmp=min(tmp,dp[i-][t][s-]);
                    }
                    if(tmp==INF) continue;
                    else dp[i][j][s]=tmp+p[i][j];
                }
            }
        }

        else
        {
            for(int s=;s<=i;s++)
            {
                LL tmp=INF;
                tmp=min(tmp,dp[i-][c[i]][s]);
                for(int t=;t<=m;t++)
                {
                    if(t==c[i]) continue;
                    tmp=min(tmp,dp[i-][t][s-]);
                }

                if(tmp==INF) continue;
                dp[i][c[i]][s]=tmp;
            }
        }
    }

    LL ans=INF;
    for(int j=;j<=m;j++) ans=min(ans,dp[n][j][k]);

    if(ans==INF) printf("-1\n");
    else printf("%lld\n",ans);

    return ;
}