LintCode:数字三角形

描述:

给定一个数字三角形,找到从顶部到底部的最小路径和。每一步可以移动到下面一行的相邻数字上。

样例:
比如,给出下列数字三角形:
[
     [2],
    [3,4],
   [6,5,7],
  [4,1,8,3]
]

从顶到底部的最小路径和为11 ( 2 + 3 + 5 + 1 = 11)。

分析:注意给出的测试用例每行的长度是不同的,之前没意识到这点,测试的时候老是提醒我数组越界

public class Solution {
    /**
     * @param triangle: a list of lists of integers
     * @return: An integer, minimum path sum
     */
    public int minimumTotal(int[][] triangle) {
        if(triangle.length<2) return triangle[0][0];
        int[][] dp=new int[triangle.length][triangle[triangle.length-1].length];
        dp[0][0]=triangle[0][0];
        for(int i=1; i<triangle.length; i++){
            dp[i][0]=dp[i-1][0]+triangle[i][0];
        }
        for(int i=1; i<triangle.length; i++){
            for(int j=1; j<triangle[i].length; j++){
                if(j!=i){
                    dp[i][j]=Math.min(dp[i-1][j]+triangle[i][j], dp[i-1][j-1]+triangle[i][j]);
                }
                else{
                    dp[i][j]=dp[i-1][j-1]+triangle[i][j];
                }
            }
        }
        int res=dp[dp.length-1][0];
        for(int j=1; j<dp[0].length; j++){
            if(res>dp[dp.length-1][j]){
                res=dp[dp.length-1][j];
            }
        }
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/karute/article/details/79949581