120. 三角形最小路径和 中等
给定一个三角形 triangle ,找出自顶向下的最小路径和。
每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说,如果正位于当前行的下标 i ,那么下一步可以移动到下一行的下标 i 或 i + 1 。
示例 1:
输入:triangle = [[2],[3,4],[6,5,7],[4,1,8,3]]
输出:11
解释:如下面简图所示:
2
3 4
6 5 7
4 1 8 3
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
示例 2:
输入:triangle = [[-10]]
输出:-10
提示:
- 1 <= triangle.length <= 200
- triangle[0].length == 1
- triangle[i].length == triangle[i - 1].length + 1
- -104 <= triangle[i][j] <= 104
进阶:
- 你可以只使用 0(n) 的额外空间(n 为三角形的总行数)来解决这个问题吗?
代码参考:
package main
import "fmt"
func main() {
triangle := [][]int{
{2},
{3, 4},
{6, 5, 7},
{4, 1, 8, 3},
// {-1},
// {2, 3},
// {1, -1, -1},
}
fmt.Println(minimumTotal(triangle)) // 2 3 5 1
}
// 和 64 题最小路径和很像,都是用简单的动态规划实现
func minimumTotal(triangle [][]int) int {
r := len(triangle)
if r <= 0 {
return 0
}
// 向下构建
steps := make([][]int, r)
for i := range triangle {
steps[i] = make([]int, i+1)
}
copy(steps, triangle)
for i := 1; i < r; i++ {
for j := 0; j <= i; j++ { // 第 N 行有 N 个元素
steps[i][j] += min(steps[i-1], j)
}
}
minStep := steps[r-1][0]
for _, step := range steps[r-1] {
if minStep > step {
minStep = step
}
}
return minStep
}
// 获取某行指定位置及其左右的最小值
func min(nums []int, c int) (minNum int) {
n := len(nums)
l, mid := c-1, c
inited := false
if 0 <= l && l <= n-1 {
inited = true
minNum = nums[l]
}
if 0 <= mid && mid <= n-1 {
if inited {
if minNum > nums[mid] {
minNum = nums[mid]
}
} else {
inited = true
minNum = nums[mid]
}
}
return
}
最后编辑: kuteng 文档更新时间: 2021-06-05 10:16 作者:kuteng