Python矩阵的最小路径和

题目：

给定一个矩阵m， 从左上角开始每次只能向右或者向下走，最后到达右下角的位置，路径上所有的数字累加起来就是路径和，返回所有的路径中最小的路径和。

 思路：

这是经典的动态规划题。矩阵m的大小为N行M列。

假设m如下：

1	3	5	9
8	1	3	4
5	0	6	1
8	8	4	0

首先生成一个大小和m一样的矩阵dp，dp中的每个元素用于记录从(0,0)位置到达该位置的最小路径和。对于dp[i][j]，能到达该位置的只有dp[i][j-1]和dp[i-1][j]，即该位置的左边或者上面，所以dp[i][j]的值为dp[i][j-1]和dp[i-1][j]中最小的那个值加上m[i][j]，再细分一下，dp[i][j]可分三种情况：

1. 第一行，i=0，此时dp[0][j]不存在上一行，所以dp[0][j]直接为前面一列的dp[0][j-1]加上m[0][j]；

1	4	9	18

2. 第一列，j=0，此时dp[i][0]不存在左边的列，所以dp[i][0]直接为上面一行的dp[i-1][0]加上m[i][0]；

1
9
14
22

3. 除了第一个和第一列之外，直接用dp[i][j-1]和dp[i-1][j]中最小的那个值加上m[i][j]即可。

1	4	9	18
9	5	8	12
14	5	11	12
22	13	15	12

代码：

下面用python实现上面的过程，为了方便查看dp加入了打印。

def min_path_sum(m):
    if m == None or len(m) == 0 or m[0] == None or len(m[0]) == 0:
        return 0
    row = len(m)
    col = len(m[0])
    dp = [[0]*col for i in range(row)]
    print('初始化的dp为：', dp)
    dp[0][0] = m[0][0]
    # 设置第一列
    for i in range(0, row):
        dp[i][0] = dp[i-1][0] + m[i][0]
    # 设置第一行
    for j in range(0, col):
        dp[0][j] = dp[0][j-1] + m[0][j]
    print('设置完第一行和第一列后，dp为：', dp)
    # 除了第一行和第一列的其它位置
    for i in range(1, row):
        for j in range(1, col):
            dp[i][j] = min(dp[i][j-1], dp[i-1][j]) + m[i][j]
    print('最终的dp为：', dp)
    return dp[row-1][col-1]

输入m，调用上面的方法查看结果：

m = [[1, 3, 5, 9], [8, 1, 3, 4], [5, 0, 6, 1], [8, 8, 4, 0]]
res = min_path_sum(m)
print('最小路径和为：', res)

结果如下：

初始化的dp为： [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
设置完第一行和第一列后，dp为： [[1, 4, 9, 18], [9, 0, 0, 0], [14, 0, 0, 0], [22, 0, 0, 0]]
最终的dp为： [[1, 4, 9, 18], [9, 5, 8, 12], [14, 5, 11, 12], [22, 13, 15, 12]]
最小路径和为： 12

 

参考：《程序员代码面试指南》左程云。对于本题，书中提到了可以降低空间复杂度的方法，感兴趣的朋友可以去看一下。