天天看点

【leetcode】#数组【Python】120. Triangle 三角形最小路径和

链接:

https://leetcode-cn.com/problems/triangle/

题目:

给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[
     [2],
    [3,4],
   [6,5,7],
  [4,1,8,3]
]
           

自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。

说明:如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。

我的解法:在triangle矩阵里用min(triangle[i-1][j-1], triangle[i-1][j]) + triangle[i][j] 赋值给triangle[i][j],最后最后一行的最小值即为所求。

class Solution(object):
    def minimumTotal(self, triangle):
        if not triangle:
            return 0
        if len(triangle) == 1:
            return triangle[0][0]
        for i in range(1, len(triangle)):
            for j in range(len(triangle[i])):
                if 0 < j < len(triangle[i])-1:
                    triangle[i][j] = min(triangle[i-1][j-1], triangle[i-1][j]) + triangle[i][j]
                elif j == 0:
                    triangle[i][j] = triangle[i-1][j] + triangle[i][j]
                else:
                    triangle[i][j] = triangle[i-1][-1] + triangle[i][j]
        return min(triangle[-1])        
           

一个意思,写了另一种写法:

class Solution(object):
    def minimumTotal(self, triangle):
        if not triangle:
            return 0
        if len(triangle) == 1:
            return triangle[0][0]
        for i in range(1, len(triangle)):
            for j in range(i+1):
                left = 1e10 if j == 0 else triangle[i-1][j-1]
                right = 1e10 if j == i else triangle[i-1][j]
                triangle[i][j] += min(left, right)
        return min(triangle[-1])