codeforces 711C Coloring Trees（DP）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf=1e15;
int c[105];
int p[105][105];
ll dp[105][105][105];
int main()
{
int n,m,b;
scanf("%d%d%d",&n,&m,&b);
for(int i=1;i<=n;i++)
scanf("%d",&c[i]);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
scanf("%d",&p[i][j]);
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=b;j++)
{
for(int k=1;k<=m;k++)
dp[i][j][k]=inf;
}
}
for(int i=1;i<=n;i++)
{
if(c[i])
{
for(int j=1;j<=b&&j<=i;j++)
{
int k=c[i];
ll min1=inf,min2=inf;
if(k>1&&j>1)
min1=*min_element(dp[i-1][j-1]+1,dp[i-1][j-1]+k);
if(k<m&&j>1)
min2=*min_element(dp[i-1][j-1]+k+1,dp[i-1][j-1]+m+1);
dp[i][j][k]=min(dp[i-1][j][k],min(min1,min2));
}
}
else
{
for(int j=1;j<=b&&j<=i;j++)
{
for(int k=1;k<=m;k++)
{
ll min1=inf,min2=inf;
if(k>1&&j>1)
min1=*min_element(dp[i-1][j-1]+1,dp[i-1][j-1]+k);
if(k<m&&j>1)
min2=*min_element(dp[i-1][j-1]+k+1,dp[i-1][j-1]+m+1);
dp[i][j][k]=p[i][k]+min(dp[i-1][j][k],min(min1,min2));
}
}
}
}
ll ans=*min_element(dp
[b]+1,dp
[b]+m+1);
if(ans<inf)
printf("%I64d\n",ans);
else
printf("-1\n");
return 0;
}