120. Triangle

java 解法 阶梯形式 动态规划保存 每个位置取上面 或者斜上面的数 重叠则需要把小的保存到flag 中 然后到最后一层则直接看那个数最小 则为最小路径

class Solution {
//消耗的空间过大
    public int minimumTotal(List> triangle) {
        int ans = Integer.MAX_VALUE;
        int flag[][] = new int[triangle.size()][triangle.size()];
        flag[0][0] = triangle.get(0).get(0);
        for (int i = 1; i < triangle.size(); i++) {
            for (int j = 0; j < triangle.get(i).size(); j++)
                flag[i][j] = Integer.MAX_VALUE;
            for (int j = 0; j < triangle.get(i).size(); j++) {
                if (j - 1 >= 0)
                    flag[i][j] = triangle.get(i).get(j) + flag[i - 1][j-1];
                if(j != triangle.get(i).size()-1)
                    flag[i][j] = Math.min(flag[i][j], triangle.get(i).get(j) + flag[i - 1][j]);
            }
        }
        int x = triangle.size();
        for(int j = 0 ; j < x ; j++) {
            ans = Math.min(flag[x-1][j], ans);
        }
        return ans;
    }
}

优化:

//空间优化: 每次计算一层(i,j)位置的数到底部的最小距离和时,当前层只跟它的下一层有关系,所以我们只用保留下一层的结果
        //就可以,用一维数组,一直更新这个一维数组就可以
class Solution {
    public int minimumTotal(List> triangle) {
    int row=triangle.size();
        int column=triangle.get(row-1).size();//最后一行数最多,作为数组列数来用
        int dp[]=new int[column];
        for(int i=0;i<column;i++)
            dp[i]=triangle.get(row-1).get(i);//最后一行到底部的距离就等于它自身的值
        for(int i=row-2;i>=0;i--) {
            for(int j=0;j<=i;j++) {
                dp[j]=Math.min(dp[j],dp[j+1])+triangle.get(i).get(j);
            }
        }
        return dp[0];
}
}

python

class Solution:
    def minimumTotal(self, triangle: List[List[int]]) -> int:
        flag = [None]* len(triangle)
        n = len(triangle)-1
        for i in range(len(triangle)):
            flag [i] = triangle[n][i]
        n = n - 1
        while n >= 0 :
            for j in range(len(triangle[n])):
                temp = flag[j]
                if(j < len(triangle[n])):
                    temp = min(temp, flag[j+1])
                flag[j] = temp + triangle[n][j]
            n = n - 1 
        return flag[0]
全部评论

相关推荐

美团 后端开发 总包n(15%是股票)
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务