题目:给定一个三角形,每一步只能移动到下一行中相邻的结点上,求出自顶向下的最小路径和。
如:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
解题思路:
可以观察到,设三角形的行数为len(triangle),dp为三角形的最后一行,从倒数第二行向上,i为当前的行数,取值范围为range(n-2,-1,-1),设j为当前行数其中的某一元素索引,则j的取值范围为range(0,i+1),可以推理得,dp[j] = min(dp[j],dp[j+1]) + dp[i][j],最后返回dp第一个值。
代码:
def square(triangle): #三角形最小路径和 n = len(triangle) #初始化dp为三角形最后一行 dp = triangle[-1] #从下(倒数第二层)到上 for i in range(n-2, -1, -1): print(i) for j in range(i+1): dp[j] = min(dp[j],dp[j+1])+triangle[i][j] return dp[0]