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

    题目链接:

    C. Coloring Trees

    time limit per test
    2 seconds
    memory limit per test
    256 megabytes
    input
    standard input
    output
    standard output

    ZS the Coder and Chris the Baboon has arrived at Udayland! They walked in the park where n trees grow. They decided to be naughty and color the trees in the park. The trees are numbered with integers from 1 to n from left to right.

    Initially, tree i has color ci. ZS the Coder and Chris the Baboon recognizes only m different colors, so 0 ≤ ci ≤ m, where ci = 0 means that tree i is uncolored.

    ZS the Coder and Chris the Baboon decides to color only the uncolored trees, i.e. the trees with ci = 0. They can color each of them them in any of the m colors from 1 to m. Coloring the i-th tree with color j requires exactly pi, j litres of paint.

    The two friends define the beauty of a coloring of the trees as the minimum number of contiguous groups (each group contains some subsegment of trees) you can split all the n trees into so that each group contains trees of the same color. For example, if the colors of the trees from left to right are 2, 1, 1, 1, 3, 2, 2, 3, 1, 3, the beauty of the coloring is 7, since we can partition the trees into 7 contiguous groups of the same color : {2}, {1, 1, 1}, {3}, {2, 2}, {3}, {1}, {3}.

    ZS the Coder and Chris the Baboon wants to color all uncolored trees so that the beauty of the coloring is exactly k. They need your help to determine the minimum amount of paint (in litres) needed to finish the job.

    Please note that the friends can't color the trees that are already colored.

    Input

    The first line contains three integers, nm and k (1 ≤ k ≤ n ≤ 100, 1 ≤ m ≤ 100) — the number of trees, number of colors and beauty of the resulting coloring respectively.

    The second line contains n integers c1, c2, ..., cn (0 ≤ ci ≤ m), the initial colors of the trees. ci equals to 0 if the tree number i is uncolored, otherwise the i-th tree has color ci.

    Then n lines follow. Each of them contains m integers. The j-th number on the i-th of them line denotes pi, j (1 ≤ pi, j ≤ 109) — the amount of litres the friends need to color i-th tree with color jpi, j's are specified even for the initially colored trees, but such trees still can't be colored.

    Output

    Print a single integer, the minimum amount of paint needed to color the trees. If there are no valid tree colorings of beauty k, print  - 1.

    Examples
    input
    3 2 2
    0 0 0
    1 2
    3 4
    5 6
    output
    10
    input
    3 2 2
    2 1 2
    1 3
    2 4
    3 5
    output
    -1
    input
    3 2 2
    2 0 0
    1 3
    2 4
    3 5
    output
    5
    input
    3 2 3
    2 1 2
    1 3
    2 4
    3 5
    output
    0

    题意:

    给这些树染色,使beauty值为k(就是分成了k段),且花费最小;

    思路:

    dp[i][j][x]表示i-1个染的色是j,前i个分成了x段的最小花费,然后就是转移了;枚举前一个的颜色和当前的颜色以及分成了多少段,要是颜色固定了就直接转移这一个颜色就好;

    AC代码:
    #include <iostream>
    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    #include <cmath>
    #include <bits/stdc++.h>
    #include <stack>
    #include <map>
     
    using namespace std;
     
    #define For(i,j,n) for(int i=j;i<=n;i++)
    #define mst(ss,b) memset(ss,b,sizeof(ss));
     
    typedef  long long LL;
     
    template<class T> void read(T&num) {
        char CH; bool F=false;
        for(CH=getchar();CH<'0'||CH>'9';F= CH=='-',CH=getchar());
        for(num=0;CH>='0'&&CH<='9';num=num*10+CH-'0',CH=getchar());
        F && (num=-num);
    }
    int stk[70], tp;
    template<class T> inline void print(T p) {
        if(!p) { puts("0"); return; }
        while(p) stk[++ tp] = p%10, p/=10;
        while(tp) putchar(stk[tp--] + '0');
        putchar('
    ');
    }
     
    const LL mod=1e9+7;
    const double PI=acos(-1.0);
    const LL inf=1e18;
    const int N=1e5+20;
    const int maxn=4e3+220;
    const double eps=1e-12;
    
    int n,m,k,c[110];
    LL p[110][110];
    LL dp[110][100][110];
    int main()
    {
       read(n);read(m);read(k);
       For(i,1,n)read(c[i]);
       For(i,1,n)For(j,1,m)read(p[i][j]);
       for(int i=1;i<=n;i++)
       {
            for(int j=1;j<=m;j++)
            {
                for(int x=1;x<=n;x++)
                    dp[i][j][x]=inf;
            }
       }
       if(c[1])dp[1][c[1]][1]=0;
       else 
       {
            for(int i=1;i<=m;i++)
            {
                dp[1][i][1]=p[1][i];
            }
       }
       for(int i=2;i<=n;i++)
       {
            if(c[i])
            {
                for(int x=1;x<=m;x++)
                {
                    for(int u=1;u<=i;u++)
                    {
                        if(dp[i-1][x][u]>=0)
                        {
                            if(x!=c[i])dp[i][c[i]][u+1]=min(dp[i][c[i]][u+1],dp[i-1][x][u]);
                            else dp[i][c[i]][u]=min(dp[i][c[i]][u],dp[i-1][x][u]);
                        }
                    }
                }
            }
            else 
            {
                for(int j=1;j<=m;j++)
                {
                    for(int x=1;x<=m;x++)
                    {
                        for(int u=1;u<=i;u++)
                        {
                            if(dp[i-1][x][u]>=0)
                            {
                                if(j!=x)dp[i][j][u+1]=min(dp[i][j][u+1],dp[i-1][x][u]+p[i][j]);
                                else dp[i][j][u]=min(dp[i][j][u],dp[i-1][x][u]+p[i][j]);
                            }
                        }
                    }
                }
    
            }
       }
       LL ans=inf;
       if(c[n])ans=dp[n][c[n]][k];
       else 
       {
            for(int i=1;i<=m;i++)if(dp[n][i][k]>=0)ans=min(ans,dp[n][i][k]);
       }
        if(ans==inf)ans=-1;
        cout<<ans<<endl;
        return 0;
    }
    

      

  • 相关阅读:
    精选的一些《编程之美》相关资料
    使用SftpDrive+SourceInsight阅读开源代码
    malloc()参数为0的情况
    《编程之美》4.5磁带文件存放优化:最优解是怎样炼成的
    从《编程之美》买票找零问题说起,娓娓道来卡特兰数——兼爬坑指南
    《编程之美》3.6判断链表是否相交之扩展:链表找环方法证明
    解答《编程之美》1.18问题1:给所有未标识方块标注有地雷概率
    C语言中 Float 数据结构的存储计算
    C#之内存分配
    unity----------------3D模型讲解
  • 原文地址:https://www.cnblogs.com/zhangchengc919/p/5820661.html
Copyright © 2011-2022 走看看