LeetCode 120. 三角形最小路径和 经典DP

  • 题意:
    从一个三角形数组的顶点出发,每次只能向下( row + 1) 或者 右下(row+1, column+ 1)。求到底部最短的路径和。

  • 思路
    是一个经典的DP入门题。开始的时候我和之前一道题搞混了,想用bfs,把和都加到最后一行,但是写完了才发现根本就不一样。。虽然一开始就知道是DP,但是还想用dfs试试,结果虽然做出来了,但是在第43个样例就超时了。还是老老实实dp吧。
    首先dp[i][j] 我们设其为,从 triangle[I][j]到底部最短的路径和,显然,我们可以得知,dp[i][j] = min(dp[i-1][j], dp[i-1][j-1]) + triangle[I][j]. 即状态转移方程。
    特殊情况是 j = 0时,只能从dp[i-1][j]走到;j = i时,只能从dp[i-1][j-1]走到。
    边界条件则是 dp[0][0] = triangle[0][0].

  • 代码:

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

Python版本:(本人Python初学,许多高级数据结构不会用)

class Solution:
    def minimumTotal(self, triangle: List[List[int]]) -> int:
        m = len(triangle)
        dp = [[0] * m for _ in range(m)]
        dp[0][0] = triangle[0][0]
        for i in range(1, m):
            dp[i][0] = dp[i-1][0] + triangle[i][0]
            for j in range(1, i):
                dp[i][j] = min(dp[i-1][j], dp[i-1][j-1]) + triangle[i][j]
            dp[i][i] = dp[i-1][i-1] + triangle[i][i]
        ans = 10**9
        for i in range(m):
            ans = min(ans, dp[m-1][i])
        return ans
  • 收获:
    经典的动态规划,要学会找状态,转移方程,边界条件。

猜你喜欢

转载自blog.csdn.net/qq_39763472/article/details/107338959