Triangle - leetcode 120号题目个人题解

题目要求

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).

解题思路

其实这是一个很简单的动态规划问题
使用一个二维的向量，cnt[i][j]表示到达第i层第j个点所需要的最小的距离。要不就是从左边下来，要不就是从右边下来，选取最短的一边再加上当前点的数值就是到达该点的最短距离，于是有如下状态转移方程:
cnt[i][j] = min(cnt[i-1][j-1],cnt[i-1][j]) + triangle[i][j]
代码实现如下：

代码实现

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
      vector<vector<int>> cnt;
      for(int i = 0;i<triangle.size();i++){
          vector<int> temp;
        for(int j = 0;j<triangle[i].size();j++){
            temp.push_back(0);
        }
        cnt.push_back(temp);
      }
      cnt[0][0] = triangle[0][0];
      for(int i = 1;i<triangle.size();i++){
        for(int j = 0;j<triangle[i].size();j++){
            if(j == 0){
                cnt[i][j] = cnt[i-1][j] + triangle[i][j];
            }
            else if(j == triangle[i].size()-1){
                cnt[i][j] = cnt[i-1][j-1] + triangle[i][j];
            }
            else{
                cnt[i][j] = min(cnt[i-1][j-1],cnt[i-1][j]) + triangle[i][j];
            }
        }
      }
      int ans = 99999;
      int level = triangle.size();
      for(int i = 0;i<level;i++){
          cout<<cnt[level-1][i]<<" ";
        if(cnt[level-1][i]<=ans){
            ans = cnt[level-1][i];
        }
      }
      return ans;
    }
};

复杂度分析

该算法的复杂度应该是O(n^2)的。