题目描述:
给你一个整数方阵 arr ,定义「非零偏移下降路径」为:从 arr 数组中的每一行选择一个数字,且按顺序选出来的数字中,相邻数字不在原数组的同一列。
请你返回非零偏移下降路径数字和的最小值。
示例 1:
输入:arr = [[1,2,3],[4,5,6],[7,8,9]]
输出:13
解释:
所有非零偏移下降路径包括:
[1,5,9], [1,5,7], [1,6,7], [1,6,8],
[2,4,8], [2,4,9], [2,6,7], [2,6,8],
[3,4,8], [3,4,9], [3,5,7], [3,5,9]
下降路径中数字和最小的是 [1,5,7] ,所以答案是 13 。
提示:
1 <= arr.length == arr[i].length <= 200
-99 <= arr[i][j] <= 99
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-falling-path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一开始是回溯,很明显200的话会超时,看代码:
class Solution {
int result = Integer.MAX_VALUE;
public int minFallingPathSum(int[][] arr) {
int len = arr.length;
// 用于记录之前的保存在哪一列中
// Set<Integer> set = new HashSet<>();
get(0,arr,0,-1);
return result;
}
public void get(int cur,int [][]arr,int tem,int pre){
if(cur == arr.length){
result = Math.min(tem,result);
return;
}
for (int i = 0; i < arr.length; i++) {
if(pre == i)
continue;
int tems = pre;
pre = i;
get(cur + 1,arr,tem + arr[cur][i],pre);
pre = tems;
}
}
}
超时,后面使用dp,动态规划
比较简单
class Solution {
int result = Integer.MAX_VALUE;
public int minFallingPathSum(int[][] arr) {
int len = arr.length;
int [][]dp = new int[len][len];
dp[0] = Arrays.copyOf(arr[0],arr.length);
for (int i = 1; i < dp.length; i++) {
for (int j = 0; j < dp.length; j++) {
dp[i][j] = Integer.MAX_VALUE;
for (int k = 0; k < dp.length; k++) {
if(k == j)
continue;
dp[i][j] = Math.min(dp[i][j],dp[i-1][k]);
}
dp[i][j] += arr[i][j];
}
}
int[] ints = dp[dp.length - 1];
for (int anInt : ints) {
result = Math.min(anInt,result);
}
return result;
}
}