关闭

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

230人阅读 评论(0) 收藏 举报
分类:

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;

}










0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:216308次
    • 积分:8820
    • 等级:
    • 排名:第2263名
    • 原创:734篇
    • 转载:4篇
    • 译文:0篇
    • 评论:27条
    233
    文章分类
    最新评论