关闭

codeforces 711C Coloring Trees(DP)

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

题目链接:http://codeforces.com/problemset/problem/711/C

O(n^4)的复杂度,以为会超时的

思路:dp[i][j][k]表示第i棵数用颜色k涂完后beauty为j

#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[n][b]+1,dp[n][b]+m+1);
    if(ans<inf)
        printf("%I64d\n",ans);
    else
        printf("-1\n");
    return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:11298次
    • 积分:840
    • 等级:
    • 排名:千里之外
    • 原创:76篇
    • 转载:2篇
    • 译文:0篇
    • 评论:0条