Triangle

来源:互联网 发布:mac快捷键的图形符号 编辑:程序博客网 时间:2024/06/06 02:43

原题链接:https://leetcode.com/problems/triangle/description/

Description: Given a triangle, find the minimum path sum from top to bottom. Each step you may move to adjacent numbers on the row below.

Solution: 我们可以采用动态规划的方法,用一个二维数组minimum[i][j]表示从triangle[i][j]走到triangle底部的最短路径。这样的状态转移方程是:minimum[i][j] = triangle[i][j] + min(minimum[i+1][j], minimum[i+1][j+1])。

这样,我们只需要从底向上进行遍历即可完成动态规划的过程,最短路径长度为minimum[0][0]。但因为minimum数组是二维的,所以需要O(n^2)的空间。

为此,我们可以将minimum数组减至一维。可以发现,从底向上的动态规划过程中,每一次状态转移后,minimum[i+1][j], minimum[i+1][j+1]都不会再用到。所以,我们可以修改状态转移方程为:minimum[j]= triangle[i][j] + min(minimum[j], minimum[j+1]),其中minimum[j]表示经triangle最后一行第j个元素结束的最短路径长度。这样只需要O(n)的空间。

class Solution {public:    int minimumTotal(vector<vector<int>>& triangle) {        int row = triangle.size();        if (row == 0) return 0;        int n = triangle[row - 1].size();        int minimum[n];        for (int i = 0; i < n; ++i) minimum[i] = triangle[row - 1][i];        for (int i = row - 2; i >= 0; --i) {            for (int j = 0; j <= i; ++j)                minimum[j] = triangle[i][j] + min(minimum[j], minimum[j + 1]);        }        return minimum[0];    }};

复杂度: 当triangle有n行时,时间复杂度为O(n^2),空间复杂度为O(n)。