Leetcode 120. 三角形最小路徑和

題目描述:
給定一個三角形,找出自頂向下的最小路徑和。每一步只能移動到下一行中相鄰的結點上。

例如,給定三角形:

[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自頂向下的最小路徑和爲 11(即,2 + 3 + 5 + 1 = 11)。

說明:

如果你可以只使用 O(n) 的額外空間(n 爲三角形的總行數)來解決這個問題,那麼你的算法會很加分。

解法1:二維dp, 自頂向下,時間複雜度和空間複雜度都是O(n^2)

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

解法2:動態規劃,自底向上,時間複雜度O(n^2), 空間複雜度O(n).

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        if(triangle.size()<=0) {
            return 0;
        }
        vector<int> dp=triangle[triangle.size()-1];
        for (int  i = triangle.size()-2; i >=0; i--)
        {
            for (int j = 0; j < triangle[i].size(); j++)
            {
                dp[j]=min(dp[j],dp[j+1])+triangle[i][j];
            }
        }
        return dp[0];
    }
};
發佈了94 篇原創文章 · 獲贊 10 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章