数据结构与算法之LeetCode-1289-下降路径最小和II 使用动态规划

最新推荐文章于 2023-08-10 10:45:54 发布

Three_ST

最新推荐文章于 2023-08-10 10:45:54 发布

阅读量128

点赞数

分类专栏：计算机基础 leetcode 笔记文章标签： leetcode 动态规划算法

本文链接：https://blog.csdn.net/qq_25482087/article/details/128403398

版权

计算机基础同时被 3 个专栏收录

50 篇文章 0 订阅

订阅专栏

leetcode

49 篇文章 0 订阅

订阅专栏

笔记

35 篇文章 0 订阅

订阅专栏

1289. 下降路径最小和 II - 力扣（LeetCode）

hard

动态规划

f[i][j] = f[i - 1][jmin[i - 1]] + arr[i][j]    其中 j != jmin[i - 1]
f[i][j] = f[i - 1][jnext[i - 1]] + arr[i][j]   其中 j == jmin[i - 1]
f[0][j] = arr[0][j]

简化

f[i][j] = first_sum + arr[i][j]    其中 j != first_pos
f[i][j] = second_sum + arr[i][j]   其中 j == first_pos

function minFallingPathSum(arr){
  let n = arr.length;
  
  let first_sum = 0, // max
      first_pos = -1,
      second_sum = 0; // second max
  for(let i=0;i<n;++i){
    let fs = Integer.MAX_VALUE, fp = -1, ss = Integer.MAX_VALUE;
    for(let j=0;j<n;++j){
      let cur_sum = (first_pos!=j?first_sum:second_sum)+arr[i][j];
      if(cur_sum<fs){
        ss = fs
        fs = cur_sum;
        fp = j;
      }else if(cur_sum<ss){
      	ss = cur_sum;
      }
    }
    first_sum = fs
    first_pos = fp
    second_sum = ss
  }
  return first_sum
}