牛客NC59-矩阵的最小路径

题目:

给定一个 n * m 的矩阵 a,从左上角开始每次只能向右或者向下走,最后到达右下角的位置,路径上所有的数字累加起来就是路径和,输出所有的路径中最小的路径和。
代码1的结果:运行超时

# @param matrix int整型二维数组 the matrix
# @return int整型
#
class Solution:
    def minPathSum(self , matrix ):
        # write code here
        if len(matrix) == 0:
            return 0
        m , n = len(matrix)-1 , len(matrix[0])-1
        return self.distance(m, n,matrix)
    
    def distance(self,x,y,matrix):
        if x != 0 and y != 0:
            return matrix[x][y] + min(self.distance(x-1, y,matrix),self.distance(x, y-1,matrix))
        if x == 0 and y != 0:
            return matrix[x][y]+self.distance(x, y-1,matrix)
        if y == 0 and x != 0:
            return matrix[x][y]+self.distance(x-1, y,matrix)
        if y == 0 and x == 0:
            return matrix[0][0]

代码2的结果:运行成功

# @param matrix int整型二维数组 the matrix
# @return int整型
#
class Solution:
    def minPathSum(self , matrix ):
        # write code here
        if len(matrix) == 0:
            return 0
        m , n = len(matrix) , len(matrix[0])
        dis_matrix = [[0]*n for i in range(m)]
        dis_matrix[0][0] = matrix[0][0]
        #第一行
        for i in range(1,n):
            dis_matrix[0][i] = dis_matrix[0][i-1] + matrix[0][i]
        #第一列
        for j in range(1,m):
            dis_matrix[j][0] = dis_matrix[j-1][0] + matrix[j][0]
        for i in range(1,n):
            for j in range(1,m):
                dis_matrix[j][i] = min(dis_matrix[j-1][i],dis_matrix[j][i-1]) + matrix[j][i]
        return dis_matrix[m-1][n-1]
     

你可能感兴趣的:(牛客,python,leetcode,算法)