1、題目
931. 下降路徑最小和
2、題解
定義一個保存狀態的二維數組，第一行初始化為A[0]，接著剩下行滿足狀態方程:
dp[i][j] =A[i][j] + min(dp[i-1][j-1],dp[i-1][j],dp[i-1][j+1]);分三種情況，當為第一列時，當為最后一列時，當為中間列時。最后取狀態數組最后一行中的最小值。
3、代碼
//931
class Solution {
public int minFallingPathSum(int[][] A) {
if (A.length == 0 || A[0].length == 0) {
return 0;
}
int[][] res = new int[A.length][A[0].length];
for(int i = 0; i < A[0].length; i++) {
res[0][i] = A[0][i];
}
for (int i = 1; i < A.length; i++) {
for (int j = 0; j < A[i].length; j++) {
if (j == 0) {
res[i][j] = Math.min(res[i - 1][j + 1], res[i - 1][j]) + A[i][j];
} else if (j == A[i].length - 1) {
res[i][j] = Math.min(res[i - 1][j - 1], res[i - 1][j]) + A[i][j];
} else {
res[i][j] = Math.min(Math.min(res[i - 1][j + 1], res[i - 1][j - 1]), res[i - 1][j]) + A[i][j];
}
}
}
int min = Integer.MAX_VALUE;
for(int i = 0; i < res[res.length - 1].length; i++) {
if(res[res.length - 1][i] < min) {
min = res[res.length - 1][i];
}
}
return min;
}
}
4、執行結果
image.png
