给你一个 n x n 的 方形 整数数组 matrix ,请你找出并返回通过 matrix 的下降路径 的 最小和 。
下降路径 可以从第一行中的任何元素开始,并从每一行中选择一个元素。在下一行选择的元素和当前行所选元素最多相隔一列(即位于正下方或者沿对角线向左或者向右的第一个元素)。具体来说,位置 (row, col) 的下一个元素应当是 (row + 1, col - 1)、(row + 1, col) 或者 (row + 1, col + 1) 。
class Solution {
int[][] memo;
public int minFallingPathSum(int[][] matrix) {
int n = matrix.length;
memo = new int[n][n];
for(int i = 0; i < n;i++){
Arrays.fill(memo[i],66666);
}
int res = Integer.MAX_VALUE;
for(int j = 0;j < n;j++){
res = Math.min(res,dp(matrix,n-1,j));
}
return res;
}
public int dp(int[][] matrix,int i,int j){
if(i < 0 || j < 0 || i >= matrix.length || j >= matrix.length){
return 99999;
}
if(i == 0){
return matrix[i][j];
}
if(memo[i][j] != 66666){
return memo[i][j];
}
//memo[i][j]=66666
memo[i][j] = matrix[i][j] + Math.min(Math.min(dp(matrix,i-1,j-1),dp(matrix,i-1,j)),dp(matrix,i-1,j+1));
return memo[i][j];
}
}
执行用时:1 ms, 在所有 Java 提交中击败了99.86%的用户
内存消耗:42.2 MB, 在所有 Java 提交中击败了9.91%的用户
通过测试用例:49 / 49