标签:matrix int 路径 一行 vector 931 Leetcode dp size
输入为一个 n * n
的二维数组 matrix
,请你计算从第一行落到最后一行,经过的路径和最小为多少。
就是说你可以站在 matrix
的第一行的任意一个元素,需要下降到最后一行。
每次下降,可以向下、向左下、向右下三个方向移动一格。也就是说,可以从 matrix[i][j]
降到 matrix[i+1][j]
或 matrix[i+1][j-1]
或 matrix[i+1][j+1]
三个位置。
思路:
使用动态规划记录每一行的最小值
然后遍历最后一行,找到其中的最小值返回
PS:遍历时注意,dp[i][j]=min(dp[i-1][j-1],dp[i-1][j],dp[i-1][j+1]),所以要判断j-1与j+1是否越界
class Solution { public: int minFallingPathSum(vector<vector<int>>& matrix) { int m=matrix.size(); int n=matrix[0].size(); vector<vector<int>> dp(m,vector<int>(n)); //base case //第0行就是matrix本身的值 for(int i=0;i<n;++i){ dp[0][i]=matrix[0][i]; } for(int i=1;i<m;++i){ for(int j=0;j<n;++j){ int tmp=dp[i-1][j]; //注意j-1与j+1是否越界 if(j-1>=0) tmp=min(tmp,dp[i-1][j-1]); if(j+1<n) tmp=min(tmp,dp[i-1][j+1]); dp[i][j]=matrix[i][j]+tmp; } } //遍历最后一行,找出最小值 int ret=INT_MAX; for(int j=0;j<n;++j){ ret=min(ret,dp[m-1][j]); } return ret; } };
标签:matrix,int,路径,一行,vector,931,Leetcode,dp,size 来源: https://www.cnblogs.com/zl1991/p/15947352.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。