leetcode 120 三角形最小路径和

把三角形看成一个上三角或者下三角(等腰三角形),然后设置状态转移公式,当没到每行的最后一个的时候,以及每行的第一个,都可以有两种方法到达当前位置,选择最小的路径和。

class Solution:
    def minimumTotal(self, triangle: List[List[int]]) -> int:

        dp=[[0 for _ in range(len(triangle[-1]))] for i in range(len(triangle))]
        dp[0][0]=triangle[0][0]

        for i in range(1,len(triangle)):
            dp[i][0]=dp[i-1][0]+triangle[i][0]
        #dp[1][1]=triangle[0][0]+triangle[1][1]

        for i in range(1,len(triangle)):
            for j in range(1,i+1):
                if j==i:
                    dp[i][j]=dp[i-1][j-1]+triangle[i][j]
                else:
                    dp[i][j]=min(dp[i-1][j-1]+triangle[i][j],dp[i-1][j]+triangle[i][j])
        print(dp)
        return min(dp[len(triangle)-1])

空间优化,通过递减的方式实现仅使用一维数组的空间。

class Solution:
    def minimumTotal(self, triangle: List[List[int]]) -> int:

        dp=[0 for _ in range(len(triangle[-1]))]
        dp[0]=triangle[0][0]


        for i in range(1,len(triangle)):

            for j in range(i,-1,-1):
                if j==i:
                    dp[j]=dp[j-1]+triangle[i][j]
                elif j==0:
                    dp[0]=dp[0]+triangle[i][j]
                else:
                    dp[j]=min(dp[j-1]+triangle[i][j],dp[j]+triangle[i][j])

        return min(dp)
全部评论

相关推荐

敢逐云霄志:你打招呼语怎么能这么长,hr都没看下去的欲望,简明扼要说重点,就读于某某学校某某专业,26届应届毕业生,学信网可查,先后在某某公司实习过(如有),然后做过什么项目,想找一份什么样的工作,可实习几个月以上,期待您的回复。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务