cf#369-C. Coloring Trees-三维dp

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

题意很简单：

有m种颜色

给n棵树，每棵树最后有一个颜色x，或者无颜色，用0表示。

要求染成k个子区间，这些子区间是这样划分的，  连续一段颜色一样的树算做一个子区间，不同颜色的数算做不同子区间，样例说的很清楚了

给n,m,k;要求最后染成k个子区间

接下来给一个n*m的矩阵，表示第n棵树染色颜色m的花费。

nmk不超过100

dp[i][j][k]表示 前i棵树染了j个子区间 ，最后一个树是k颜色

转移很好转移  

枚举第i棵树的颜色，如果已经有颜色了，则   如果前一棵树颜色和i一样，则

                        dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j][x]);

否则

                        dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j-1][x]);

如果无颜色，则枚举 给第i棵树涂的颜色，同样按上面的思路去转移。

最坏复杂度n*m*k*k

sb题  m和k写反了找半天错误。。GGGG还是太菜了

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;

const double pi=acos(-1.0);
double eps=0.000001;
typedef long long  ll;

ll mp[105][105];
ll aa[105];
ll dp[105][105][105];
int  main()
{
    ll n,m,k;
    cin>>n>>m>>k;
    for (int i=1; i<=n; i++)
        scanf("%lld",&aa[i]);

    for (ll i=1; i<=n; i++)
        for (ll j=1; j<=m; j++)
            scanf("%lld",&mp[i][j]);

    for (ll i=0; i<=100; i++)
        for (ll j=0; j<=100; j++)
            for (ll x=0; x<=100; x++)
                dp[i][j][x]=1e16;


    if (aa[1])
        dp[1][1][aa[1]]=0;
    else
    {
        for (int i=1; i<=m; i++)
            dp[1][1][i]=mp[1][i];
    }
    for (ll i=2; i<=n; i++)
    {
        for (ll j=1; j<=i; j++)
        {
            if (aa[i])
            {
                for (ll x=1; x<=m; x++)
                {
                    if (x==aa[i])
                        dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j][x]);
                    else
                        dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j-1][x]);
                }
                continue;
            }
            for (ll x=1; x<=m; x++)     //itself
            {
                for (ll y=1; y<=m; y++)
                {
                    if (x==y)
                        dp[i][j][x]=min(dp[i][j][x],dp[i-1][j][y]+mp[i][x]);
                    else
                        dp[i][j][x]=min(dp[i][j][x],dp[i-1][j-1][y]+mp[i][x]);
                }
            }
        }
    }
    ll ans=1e16;
    for (int i=1; i<=m; i++)
        ans=min(ans,dp[n][k][i]);
    if (ans==1e16)
        printf("-1\n");
    else
        printf("%lld\n",ans);
    return 0;

}