lintcode triangle 数字三角形

问题描述

lintcode

笔记

代码1

设置buff[i][j]为：以元素(i,j)结尾的最小路径和。
buff[i][j]应该是buff[i-1][j-1]和buff[i-1][j]中的较小值再加上当前的triangle[i][j]。
则状态转移方程为：
buff[i][j] = min(buff[i-1][j-1], buff[i-1][j] + triangle[i][j]
初始条件为：
buff[0][0] = triangle[0][0]

注意

判断越界问题也用到了一个小技巧，忘了从哪学来的了。

//要取buff[i-1][j-1]，和buff[i-1][j]，但是要保证不越界。
//第i-1行的j的取值范围为[0, i-1]
int lo = max(0, j-1);//当j-1小于0的时候只取0，大于等于0的时候取自身。
int hi = min(j, i-1);//当j大于i-1的时候只取i-1，小于等于i-1的时候取自身。

代码2

题目要求优化空间复杂度。其实不需要用二维数组，因为每一行的值只与上一行的值相关，因此可以只用一行做缓存即可。（代码2）
在这里用到了背包问题中的小技巧，每一行从后往前更新，以防覆盖前面要使用的值。

注意

看来，要将二维空间节省成一维空间的时候，都要考虑一下从后往前向前更新这一小技巧。

代码1

class Solution {
public:
    /**
     * @param triangle: a list of lists of integers.
     * @return: An integer, minimum path sum.
     */
    int minimumTotal(vector<vector<int> > &triangle) {
        // write your code here
        const int len = triangle.size();
        vector<vector<int>> buff(len, vector<int>(len));
        buff[0][0] = triangle[0][0];
        for (int i = 1; i < len; i++)
        {
            for (int j = 0; j <= i; j++)
            {
                // 要取buff[i-1][j-1]，和buff[i-1][j]，但是要保证不越界。
                // 第i-1行的j的取值范围为[0, i-1]
                int lo = max(0, j-1);
                int hi = min(j, i-1);
                buff[i][j] = min(buff[i-1][lo], buff[i-1][hi]) + triangle[i][j];
            }
        }
        int res = buff[len-1][0];
        for (int i = 1; i < len; i++)
            res = min(res, buff[len-1][i]);
        return res;
    }
};

代码2

class Solution {
public:
    /**
     * @param triangle: a list of lists of integers.
     * @return: An integer, minimum path sum.
     */
    int minimumTotal(vector<vector<int> > &triangle) {
        // write your code here
        const int len = triangle.size();
        vector<int> buff(len);
        buff[0] = triangle[0][0];
        for (int i = 1; i < len; i++)
        {
            for (int j = i; j >= 0; j--)
            {
                int lo = max(0, j-1);
                int hi = min(j, i-1);
                buff[j] = min(buff[lo], buff[hi]) + triangle[i][j];
            }
        }
        int res = buff[0];
        for (int i = 1; i < len; i++)
            res = min(res, buff[i]);
        return res;
    }
};