首页 > 代码库 > leetcode 【 Triangle 】python 实现

leetcode 【 Triangle 】python 实现

题目

Given a triangle, find the minimum path sum from top to bottom. Each step you may move to adjacent numbers on the row below.

For example, given the following triangle

[     [2],    [3,4],   [6,5,7],  [4,1,8,3]]

 

The minimum path sum from top to bottom is 11 (i.e., 2 + 3 + 5 + 1 = 11).

Note:
Bonus point if you are able to do this using only O(n) extra space, where n is the total number of rows in the triangle.

 

代码:oj测试通过 Runtime: 82 ms

 1 class Solution: 2     # @param triangle, a list of lists of integers 3     # @return an integer 4     def minimumTotal(self, triangle): 5         # special case 6         if len(triangle)==0: 7             return 0 8         # dp visit 9         LEVEL = len(triangle)10         dp = [0 for i in range(LEVEL)]11         for i in range(LEVEL):12             for j in range(len(triangle[i])-1,-1,-1):13                 if j==len(triangle[i])-1 :14                     dp[j] = dp[j-1] + triangle[i][j]15                 elif j==0 :16                     dp[0] = dp[0] + triangle[i][0]17                 else:18                     dp[j] = min(dp[j-1],dp[j]) + triangle[i][j]19         return min(dp)

 

思路

典型的动态规划,思路跟Unique Path类似,详情见

http://www.cnblogs.com/xbf9xbf/p/4250359.html

另,这道题还有一个bonus,如何用尽量少的额外空间。

一般的dp思路是,定义一个O(n)的空间,跟三角形等大小的额外空间。

这里只用三角形最底层那一层的大小的空间。

遍历第i层时,利用 dp[1:len(triangle[i])] 的空间存储开始节点到第i层各个节点的最小和。

这里注意:从后往前遍历可以节省数组空间。这是Array的一个常见操作技巧,详情见

http://www.cnblogs.com/xbf9xbf/p/4240257.html

连续刷刷题,能把前后的技巧多关联起来,对代码的能力提升有一定的帮助。

leetcode 【 Triangle 】python 实现