leetcode931.下降路径最小和

题目大意

给定一个方形整数数组 A，我们想要得到通过 A 的下降路径的最小和。

下降路径可以从第一行中的任何元素开始，并从每一行中选择一个元素。在下一行选择的元素和当前行所选元素最多相隔一列。

示例：

输入：[[1,2,3],[4,5,6],[7,8,9]]
输出：12
解释：
可能的下降路径有：

• [1,4,7], [1,4,8], [1,5,7], [1,5,8], [1,5,9]
• [2,4,7], [2,4,8], [2,5,7], [2,5,8], [2,5,9], [2,6,8], [2,6,9]
• [3,5,7], [3,5,8], [3,5,9], [3,6,8], [3,6,9]

和最小的下降路径是 [1,4,7]，所以答案是 12。

解题思路

常规问题，动态规划。dp[i][j]表示到达（i,j）位置时，最小代价。

class Solution {
public:
    int minFallingPathSum(vector<vector<int>>& A) {
    	int length = A.size();

    	vector<vector<int>> dp(length, vector<int>(length, INT_MAX));
    	// 初始化dp数组
    	for (int i = 0; i < length; ++i)
    		dp[0][i] = A[0][i];

    	for (int i = 1; i < length; ++i){
    		for (int j = 0; j < length; ++j){
    			// 防止越界
    			if (j - 1 >= 0)
    				dp[i][j] = min(dp[i][j], dp[i - 1][j - 1] + A[i][j]);
    			if (j + 1 < length)
    				dp[i][j] = min(dp[i][j], dp[i - 1][j + 1] + A[i][j]);
    			dp[i][j] = min(dp[i][j], dp[i - 1][j] + A[i][j]);
    		}
    	}

    	int res = INT_MAX;
    	for (int i = 0; i < length; ++i)
    		res = min(res, dp[length - 1][i]);

    	return res;
    }
};