zoukankan      html  css  js  c++  java
  • codeforces 711C Coloring Trees(DP)

    题目链接: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
    ",ans);
        else
            printf("-1
    ");
        return 0;
    }
    

     


  • 相关阅读:
    SpringCloud入门
    SpringBoot自动配置的演示
    SpringBoot自动配置原理
    SpringBoot整合Junit、Mybatis以及Redis
    SpringBoot介绍
    面向对象编程
    Spring MVC介绍
    Queue
    springcloud项目 报错数据库未配置
    linux环境下mongoDB主从复制搭建
  • 原文地址:https://www.cnblogs.com/westwind1005/p/5975201.html
Copyright © 2011-2022 走看看