阿里实习生面试编程题

有个穷困的艺术家。他画了一幅超现实主义的作品《方块手拉手》。现在他已经把图画中手拉手的一排大小不一的方块都画出来了。现在要考虑上颜色了。可惜他手中的钱并不多了。但是他是个有追求的人，他希望这幅画中每两个相邻的方块的颜色是不一样的。你能帮他计算一下把这幅画上色后，最少需要花多少钱么。

输入 N个方块，K个颜色

接下来每列是各个方块染成不同颜色的价钱

输出 最少花的钱

输入

4 3

2 3 2

9 1 4

7 8 1

2 8 3

输出

6

#include<iostream>  
#include<algorithm>  
#include<sstream>  
#define MAX 101  
using namespace std;  
int n, k;  
int D[MAX][MAX];  
int main()  
{   
    cin >> n >> k;//n个块，k个颜色  
    for (int i = 1; i <= n; i++)//输入价格  
        for (int j = 1; j <= k; j++)  
            cin >> D[i][j];  
    for (int i = n - 1; i >= 1; i--)//从倒数第二行开始，计算当前值与下一行中不同列的元素的最小值  
    {  
        for (int j = 1; j <= k; j++)  
        {  
            int min = 1000;  
            for (int r = 1; r <= k; r++)  
            {  
                if (r == j)  
                    continue;  
                if (D[i][j] + D[i + 1][r] < min)  
                    min = D[i][j] + D[i + 1][r];  
            }  
            D[i][j] = min;  
        }  
    }//动态规划完毕  
    int min = 1000;  
    for (int i = 1; i <= k; i++)//取出最小值  
    {  
        if (D[1][i] < min)  
            min=D[1][i];  
    }  
    cout << min;  
    return 0;  
}