Leetcode 120. Triangle

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.

题意:在一个二维数组表示的三角形中,找一个从定点到底的路径和最小的一条路径。

思路:
第一种方法是用深度优先搜索暴力的求解每个路径和,找到最短的路径和,因为每个点下面都有两条路可以走,所以时间复杂度是2的n次方级别。
第二种方法是用动态规划的思路。用dp[i][j]表示从顶点2到第i层j列位置的最短路径和,由图可知,从上层有两个位置能到达,如果知道了上层两个位置的dp[i-1][j-1]和dp[i-1][j],则dp[i][j] = Math.min(dp[i-1][j-1], dp[i-1][j]) + array[i][j]。三角形的两条腰是特殊情况,它们只能从上层一个位置到达。

public int minimumTotal(List<List<Integer>> triangle) {
    if (triangle == null || triangle.size() == 0) {
        return 0;
    }

    int len = triangle.size();
    int[][] dp = new int[len][len];
    dp[0][0] = triangle.get(0).get(0);
    int min = Integer.MAX_VALUE;
    for (int i = 1; i < len; i++) {
        for (int j = 0; j <= i; j++) {
            if (j == 0) {
                dp[i][j] = dp[i - 1][j] + triangle.get(i).get(j);
            } else if (j == i) {
                dp[i][j] = dp[i - 1][i - 1] + triangle.get(i).get(j);
            } else {
                dp[i][j] = Math.min(dp[i - 1][j - 1], dp[i - 1][j]) + triangle.get(i).get(j);
            }
            if (i == len - 1) {
                min = Math.min(min, dp[i][j]);
            }
        }
    }

    return min == Integer.MAX_VALUE ? dp[0][0] : min;
}

题目还要求能否用O(n)的空间复杂度解决,参看discuss的答案,他的思路和自己的思路是一样的,只不过是从底向顶点推到,即dp[i]代表从当前层第i列走到底的最短路径和,而我写的dp状态是从顶点到当前位置的最短路径和。

public int minimumTotal(List<List<Integer>> triangle) {
    int[] A = new int[triangle.size()+1];
    for(int i=triangle.size()-1;i>=0;i--){
        for(int j=0;j<triangle.get(i).size();j++){
            A[j] = Math.min(A[j],A[j+1])+triangle.get(i).get(j);
        }
    }
    return A[0];
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,776评论 0 33
  • 动态规划(Dynamic Programming) 本文包括: 动态规划定义 状态转移方程 动态规划算法步骤 最长...
    廖少少阅读 3,338评论 0 18
  • 原题 给定一个数字三角形,找到从顶部到底部的最小路径和。每一步可以移动到下面一行的相邻数字上。 比如,给出下列数字...
    Jason_Yuan阅读 1,333评论 0 1
  • 原创 李苗苗 高育良者,初为汉东大学教授,授法家之学。其妻吴氏,亦于汉东大学授明史,二人同寝同食,同息同止...
    余歌子_李阅读 563评论 0 1
  • 作为一个还没有真正走出象牙塔的人,越临近毕业,越质疑自己每天努力学习的必要性。总有人用“理想很丰满,现实很骨感”的...
    迟语者阅读 209评论 0 0