[leetcode] 120. Triangle

xiaoxiao2021-02-28  31

题目:

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.

思路:

虽然说题目说的是要自顶向下,但是其实自底向上的效果是一致的。状态转移是 triangle[i][j] += min(triangle[i+1][j], triangle[i+1][j+1])(这里不开辟新的空间,如果没有影响的时候可以重用triangle)

代码:C++

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

改进:参考leetcode的discuss

自顶向下:

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

自底向上:

class Solution { public: int minimumTotal(vector<vector<int>>& triangle) { int n = triangle.size(); vector<int>dp(n); for(int i = 0; i < n; i++){ if(i != 0) dp[i] = dp[i - 1] + triangle[i][i]; for(int j = i - 1; j > 0; j--) dp[j] = triangle[i][j] + min(dp[j], dp[j - 1]); dp[0] = dp[0] + triangle[i][0]; } return *min_element(dp.begin(), dp.end()); } };

转载请注明原文地址: https://www.6miu.com/read-2624444.html

最新回复(0)