Leetcode 931. Minimum falling path sum 最小下降路径和(动态规划) 题目描述
已知一个正方形二维数组A，我们想找到一条最小下降路径的和
所谓下降路径是指，从一行到下一行，只能选择间距不超过1的列（也就是说第一行的第一列，只能选择第二行的第一列和第二列；第二行的第二列，只能选择第三行的第一列第二列第三列），最小下降路径就是这个路径的和最小
测试样例 Input: [[1,2,3],[4,5,6],[7,8,9]] Output: 12 Explanation: 可能的下降路径是: [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. 详细分析随便举个例子:
| 1 | 2 | 3 |
|---|---|---|
| 4 | 5 | 6 |
| 7 | 8 | 9 |
考虑(1,0)这个格子(4)
(1,0)有dp[1][0] = 4 + min(1,2)
(1,1)有dp[1][1] = 5 + min(1,2,3)
(1,2)有dp[1][2] = 6 + min(2,3)
总结规律即可
算法实现 class Solution { public: int minFallingPathSum(vector<vector<int>>& A) { int dp[200][200]={0}; for(int i=0;i<A.size();i++){ dp[0][i]=A[0][i]; } for(int i=1;i<A.size();i++){ for(int k=0;k<A[0].size();k++){ if(k==0){ dp[i][k] = A[i][k]+ min(dp[i-1][k],dp[i-1][k+1]); }else if(k==A.size()-1){ dp[i][k] = A[i][k]+ min(dp[i-1][k],dp[i-1][k-1]); }else{ dp[i][k] = A[i][k]+ min(dp[i-1][k],min(dp[i-1][k-1],dp[i-1][k+1])); } } } return *min_element(&dp[A.size()-1][0],&dp[A.size()-1][A.size()]); } };