【第22期】观点:IT 行业加班,到底有没有价值?

Coloring Trees (三维DP)

原创 2016年08月30日 15:48:22

B题没有考虑到除了0点以外,原来就不符合的情况,又掉分了…..
题目
题意:
n颗树,每棵树可以涂m种颜色,把树分成k组,连续相同颜色的树为合并为一组
 给出了n颗树的涂色情况,当涂色为0的时候,表示还没有涂颜色
 给n * m 的矩阵,第i行j列为 第i个点,图j颜色的消耗
 求n颗树涂颜色涂成k组时,用的最少消耗是多少

思路:
状态方程:
dp[i][k][j] 表示到了第i个点,分成k组, 颜色为j时,最少消耗为多少
 当前点被涂过颜色:
dp[now][j + (a[i] != k)][a[i]] = min(dp[now][j + (a[i] != k)][a[i]], dp[next][j][k]);
当没被涂过:
枚举颜色:
 for(int l = 1; l <= m; l++)
dp[now][j + (l != k)][l] = min(dp[now][j + (l != k)][l], dp[next][j][k] + val[i][l]);

样例:
input
3 2 2
0 0 0
1 2
3 4
5 6
output
10

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <vector>
#include <cmath>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long LL;
const int maxn = 1e5 + 5;
int n, m, K, c[105], a[105][105];
LL f[105][105][105];
int main() {
    scanf("%d%d%d", &n, &m, &K);
    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", &a[i][j]);
    memset(f, 0x3f, sizeof(f));
    if (c[1]) f[1][1][c[1]] = 0;
    else for (int i=1; i<=m; i++) f[1][1][i] = a[1][i];
    for (int i=2; i<=n; i++) {
        for (int j=1; j<=i&&j<=K; j++) {
            if (c[i]) {
                f[i][j][c[i]] = min(f[i][j][c[i]], f[i-1][j][c[i]]);
                for (int k=1; k<=m; k++) if (c[i]!=k) {
                    f[i][j][c[i]] = min(f[i][j][c[i]], f[i-1][j-1][k]);
                }
            } else {
                for (int k=1; k<=m; k++) {
                    f[i][j][k] = min(f[i][j][k], f[i-1][j][k] + a[i][k]);
                    for (int q=1; q<=m; q++) if (q!=k) {
                        f[i][j][k] = min(f[i][j][k], f[i-1][j-1][q] + a[i][k]);
                    }
                }
            }
        }
    }
    LL ans = f[104][104][104];
    for (int i=1; i<=m; i++) {
        ans = min(ans, f[n][K][i]);
    }
    if (ans == f[104][104][104]) {
        cout << "-1" << endl;
    } else {
        cout << ans << endl;
    }
    return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。 举报

相关文章推荐

Codeforces #369(Div.2) C.Coloring Trees【Dp】

Coloring Trees Description ZS the Coder and Chris the Baboon has arrived at Udayland! ...

uva 193 - Graph Coloring

点击打开链接 题目意思:

Codeforces Round #369 (Div. 2) -- C. Coloring Trees (三维DP)

大体题意: 给你n 个树,你要给这些树染色,  标号是0 表示这棵树还没有染色,标号不是0 表示这棵树已经染色 不需要再染,这片树的美丽程度是  连续相同颜色的数目! 告诉你指定美丽程度K,和  最多...

193 - Graph Coloring(*****)推荐入门

/* 推荐题型:五星 题意:对图中节点进行染色,只有黑白可选,要求相邻节点不同色,问黑色最多可有多少个 思路:深度搜索、记忆数组 */ #include &lt;cstdio&gt; #include &lt;cstring&gt; const int nMax=107; int G[nMax][nMax]; int ans[nMax],res,n; //answer,result void dfs(int cur,int *A) { int color[nMax]; if(cur==n) { int num=0; fo

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

http://codeforces.com/contest/711/problem/C 题意很简单: 有m种颜色 给n棵树,每棵树最后有一个颜色x,或者无颜色,用0表示。 要求染成k个...
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)