【动态规划】Codeforces 711C Coloring Trees

题目链接：

　　http://codeforces.com/problemset/problem/711/C

题目大意：

　　给N棵树，M种颜色，已经有颜色的不能涂色，没颜色为0，可以涂色，每棵树I涂成颜色J花费P_IJ。求分成K个颜色段（1112221为3个颜色段）的最小花费。无解输出-1.

题目思路：

　　【动态规划】

　　f[i][j][k]表示前i个树分成j段，最后一个颜色是k的花费。

　　根据当前这棵树是否可以涂色，上一棵树是否可以涂色转移。

　　直接枚举这个树和上棵树的颜色，N⁴居然没T。 

//
//by coolxxx
//#include<bits/stdc++.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<iomanip>
#include<map>
#include<stack>
#include<queue>
#include<set>
#include<bitset>
#include<memory.h>
#include<time.h>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
//#include<stdbool.h>
#include<math.h>
#define min(a,b) ((a)<(b)?(a):(b))
#define max(a,b) ((a)>(b)?(a):(b))
#define abs(a) ((a)>0?(a):(-(a)))
#define lowbit(a) (a&(-a))
#define sqr(a) ((a)*(a))
#define swap(a,b) ((a)^=(b),(b)^=(a),(a)^=(b))
#define mem(a,b) memset(a,b,sizeof(a))
#define eps (1e-8)
#define J 10
#define mod 1000000007
#define MAX 0x7f7f7f7f
#define PI 3.14159265358979323
#define N 104
using namespace std;
typedef long long LL;
int cas,cass;
int n,m,lll,ans;
int c[N];
LL aans;
LL a[N][N];
LL f[N][N][N];
int main()
{
    #ifndef ONLINE_JUDGE
//    freopen("1.txt","r",stdin);
//    freopen("2.txt","w",stdout);
    #endif
    int i,j,k,l;
//    for(scanf("%d",&cass);cass;cass--)
//    for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
//    while(~scanf("%s",s+1))
    while(~scanf("%d",&n))
    {
        scanf("%d%d",&m,&cas);
        for(i=1;i<=n;i++)
            scanf("%d",&c[i]);
        for(i=1;i<=n;i++)
            for(j=1;j<=m;j++)
                scanf("%I64d",&a[i][j]);
        mem(f,0x7f);
        c[0]=m+1,f[0][0][m+1]=0;
        for(i=1;i<=n;i++)
        {
            if(c[i]!=0)
            {
                if(c[i-1]!=0)
                {
                    for(j=1;j<=min(i,cas);j++)
                    {
                        if(c[i]!=c[i-1])
                            f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j-1][c[i-1]]);
                        else f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j][c[i-1]]);
                    }
                }
                else//c[i]!=0 c[i-1]=0
                {
                    for(j=1;j<=min(i,cas);j++)
                    {
                        for(k=1;k<=m;k++)
                        {
                            if(k!=c[i])
                                f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j-1][k]);
                            else f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j][k]);
                        }
                    }
                }
            }
            else if(c[i-1]==0)//c[i]=0
            {
                for(j=1;j<=min(i,cas);j++)
                {
                    for(k=1;k<=m;k++)
                    {
                        for(l=1;l<=m;l++)
                        {
                            if(l==k)f[i][j][k]=min(f[i][j][k],f[i-1][j][l]+a[i][k]);
                            else f[i][j][k]=min(f[i][j][k],f[i-1][j-1][l]+a[i][k]);
                        }
                    }
                }
            }
            else//c[i]=0,c[i-1]!=0
            {
                for(j=1;j<=min(i,cas);j++)
                {
                    for(k=1;k<=m;k++)
                    {
                        if(k!=c[i-1])
                            f[i][j][k]=min(f[i][j][k],f[i-1][j-1][c[i-1]]+a[i][k]);
                        else f[i][j][k]=min(f[i][j][k],f[i-1][j][c[i-1]]+a[i][k]);
                    }
                }
            }
        }
        aans=f[0][0][0];
        for(i=1;i<=m;i++)
            aans=min(aans,f[n][cas][i]);
        if(aans==f[0][0][0])puts("-1");
        else printf("%I64d
",aans);
    }
    return 0;
}
/*
//

//
*/