LeetCode120. 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.

题目大意是:给定一个数字组成的三角形,要求从顶层到底层,找到一条最小路径。使得在这条路径上所有数之和是所有路径中最小的。往下一层走时,你只能访问到与当前位置相邻的两个数。

利用动态规划的思想可以很快的解决问题。对于每一层的每一个数,都可以有一个值minValue,代表“从顶层开始到这里结束的最小路径和”。而这个值显然是“当前数值”与“左上方数的minValue”或“右上方数的minValue”之和。即状态方程如下:

minValue[current] = min{ minValue[left_top], minValue[right_top] } + currentValue

于是,设置一个数组minValue用于记录最小路径和,每个点的坐标以行号row和行中的序号num表示,则其最小路径和为minValue[row * (row + 1) / 2 + num]。从顶层开始遍历,计算出每一个minValue[x]的值,最后找到底层最小的一个值,便是想要的答案。

代码实现如下:

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        int rowNum = triangle.size();
        int totalNum = rowNum * (rowNum + 1) / 2;
        int* minValue = new int[totalNum] { 0 };
        minValue[0] = triangle[0][0];
        for (int row = 1; row < rowNum; row++) {
            for (int num = 0; num < triangle[row].size(); num++) {
                int left = 200000, right = 200000;
                if (num - 1 >= 0) {
                    left = minValue[getIndex(row - 1, num - 1)] + triangle[row][num];
                }
                if (num <= row - 1) {
                    right = minValue[getIndex(row - 1, num)] + triangle[row][num];
                }
                minValue[getIndex(row, num)] = left < right ? left : right;
            }
        }
        int min = 200000;
        for (int i = 0; i < rowNum; i++) {
            if (minValue[getIndex(rowNum - 1, i)] < min) {
                min = minValue[getIndex(rowNum - 1, i)];
            }
        }
        return min;
    }
private:
    int getIndex(int row, int num) {
        return (row * (row + 1) / 2 + num);
    }
};

然而,题目中给出了一个加分项:只用O(n)的额外空间来解题(n为行数),而上述解法中,对每一个数值的minValue进行了保存,实际的额外空间为O(n*(n+1)/2)

如何只是用O(n)的额外空间呢?显然,我们在动态规划的过程中,是以行为一个单位进行的。而实际上每计算出当前行的所有minValue值,上一行的值就已经没有意义,不需要保留了。因此,实际上我们只需要一个长度为n的数组来保存数据,n是最长的行的长度,也是行数。

Discuss中的一个解法如下:

class Solution {
public:
    int minimumTotal(vector<vector<int> > &triangle) {
        int n = triangle.size();
        vector<int> minlen(triangle.back());
        for (int layer = n-2; layer >= 0; layer--) // For each layer
        {
            for (int i = 0; i <= layer; i++) // Check its every 'node'
            {
                // Find the lesser of its two children, and sum the current value in the triangle with it.
                minlen[i] = min(minlen[i], minlen[i+1]) + triangle[layer][i]; 
            }
        }
        return minlen[0];
    }
};

这里采用了从底层到底层的构造顺序。每一层记录minValue只采用了一个长度为n的数组,达成了目标。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值