(5千字)由浅入深讲解动态规划(JS版)-钢条切割,最大公共子序列,最短编辑距离

斐波拉契数列

首先咱们来看看斐波拉契数列,这是一个你们都很熟悉的数列:javascript

// f = [1, 1, 2, 3, 5, 8]
f(1) = 1;
f(2) = 1;
f(n) = f(n-1) + f(n -2); // n > 2

有了上面的公式,咱们很容易写出计算f(n)的递归代码:html

function fibonacci_recursion(n) {
  if(n === 1 || n === 2) {
    return 1;
  }
  
  return fibonacci_recursion(n - 1) + fibonacci_recursion(n - 2);
}

const res = fibonacci_recursion(5);
console.log(res);   // 5

如今咱们考虑一下上面的计算过程,计算f(5)的时候须要f(4)与f(3)的值,计算f(4)的时候须要f(3)与f(2)的值,这里f(3)就重复算了两遍。在咱们已知f(1)和f(2)的状况下,咱们其实只须要计算f(3),f(4),f(5)三次计算就好了,可是从下图可知,咱们总共计算了8次,里面f(3), f(2), f(1)都有屡次重复计算。若是n不是5,而是一个更大的数,计算次数更是指数倍增加。不考虑已知1和2的状况的话,这个递归算法的时间复杂度是$O(2^n)$。java

非递归的斐波拉契数列

为了解决上面指数级的时间复杂度,咱们不能用递归算法了,而要用一个普通的循环算法。应该怎么作呢?咱们只须要加一个数组,里面记录每一项的值就好了,为了让数组与f(n)的下标相对应,咱们给数组开头位置填充一个0算法

const res = [0, 1, 1];
f(n) = res[n];

咱们须要作的就是给res数组填充值,而后返回第n项的值就好了:数组

function fibonacci_no_recursion(n) {
  const res = [0, 1, 1];
  for(let i = 3; i <= n; i++){
    res[i] = res[i-1] + res[i-2];
  }
  
  return res[n];
}

const num = fibonacci_no_recursion(5);
console.log(num);   // 5

上面的方法就没有重复计算的问题,由于咱们把每次的结果都存到一个数组里面了,计算f(n)的时候只须要将f(n-1)和f(n-2)拿出来用就好了,由于是从小往大算,因此f(n-1)和f(n-2)的值以前就算好了。这个算法的时间复杂度是O(n),比$O(2^n)$好的多得多。这个算法其实就用到了动态规划的思想。数据结构

动态规划

动态规划主要有以下两个特色测试

  1. 最优子结构:一个规模为n的问题能够转化为规模比他小的子问题来求解。换言之,f(n)能够经过一个比他规模小的递推式来求解,在前面的斐波拉契数列这个递推式就是f(n) = f(n-1) + f(n -2)。通常具备这种结构的问题也能够用递归求解,可是递归的复杂度过高。
  2. 子问题的重叠性:若是用递归求解,会有不少重复的子问题,动态规划就是修剪了重复的计算来下降时间复杂度。可是由于须要存储中间状态,空间复杂度是增长了。

其实动态规划的难点是概括出递推式,在斐波拉契数列中,递推式是已经给出的,可是更多状况递推式是须要咱们本身去概括总结的。spa

钢条切割问题

先看看暴力穷举怎么作,以一个长度为5的钢条为例:code

上图红色的位置表示能够下刀切割的位置,每一个位置能够有切和不切两种状态,总共是$2^4 = 16$种,对于长度为n的钢条,这个状况就是$2^{n-1}$种。穷举的方法就不写代码了,下面直接来看递归的方法:htm

递归方案

仍是以上面那个长度为5的钢条为例,假如咱们只考虑切一刀的状况,这一刀的位置能够是1,2,3,4中的任意位置,那切割以后,左右两边的长度分别是:

// [left, right]: 表示切了后左边,右边的长度
[1, 4]: 切1的位置
[2, 3]: 切2的位置
[3, 2]: 切3的位置
[4, 1]: 切4的位置

分红了左右两部分,那左右两部分又能够继续切,每部分切一刀,又变成了两部分,又能够继续切。这不就将一个长度为5的问题,分解成了4个小问题吗,那最优的方案就是这四个小问题里面最大的那个值,同时不要忘了咱们也能够一刀都不切,这是第五个小问题,咱们要的答案其实就是这5个小问题里面的最大值。写成公式就是,对于长度为n的钢条,最佳收益公式是:

  • $r_n$ : 表示咱们求解的目标,长度为n的钢条的最大收益
  • $p_n$: 表示钢条彻底不切的状况
  • $r_1 + r_{n-1}$: 表示切在1的位置,分为了左边为1,右边为n-1长度的两端,他们的和是这种方案的最优收益
  • 咱们的最大收益就是不切和切在不一样状况的子方案里面找最大值

上面的公式已经能够用递归求解了:

const p = [0, 1, 5, 8, 9, 10, 17, 17, 20, 24, 30]; // 下标表示钢条长度,值表示对应价格

function cut_rod(n) {
  if(n === 1) return 1;
  
  let max = p[n];
  for(let i = 1; i < n; i++){
    let sum = cut_rod(i) + cut_rod(n - i);
    if(sum > max) {
      max = sum;
    }
  }
  
  return max;
}

cut_rod(9);  // 返回 25

上面的公式还能够简化,假如咱们长度9的最佳方案是切成2 3 2 2,用前面一种算法,第一刀将它切成2 75 4,而后两边再分别切最终均可以获得2 3 2 2,因此5 4方案最终结果和2 7方案是同样的,都会获得2 3 2 2,若是这两种方案,两边都继续切,其实还会有重复计算。那长度为9的切第一刀,左边的值确定是1 -- 9,咱们从1依次切过来,若是后面继续对左边的切割,那继续切割的那个左边值一定是咱们前面算过的一个左边值。好比5 4切割成2 3 4,其实等价于第一次切成2 7,第一次若是是3 6,若是继续切左边,切为1 2 6,其实等价于1 8,都是前面切左边为1的时候算过的。因此若是咱们左边依次是从1切过来的,那么就没有必要再切左边了,只须要切右边。因此咱们的公式能够简化为: $$ r_n = \max_{1<=i<=n}(pi+r_{n-i}) $$ 继续用递归实现这个公式:

const p = [0, 1, 5, 8, 9, 10, 17, 17, 20, 24, 30]; // 下标表示钢条长度,值表示对应价格

function cut_rod2(n) {
  if(n === 1) return 1;
  
  let max = p[n];
  for(let i = 1; i <= n; i++){
    let sum = p[i] + cut_rod2(n - i);
    if(sum > max) {
      max = sum;
    }
  }
  
  return max;
}

cut_rod2(9);  // 结果仍是返回 25

上面的两个公式都是递归,复杂度都是指数级的,下面咱们来说讲动态规划的方案。

动态规划方案

动态规划方案的公式和前面的是同样的,咱们用第二个简化了的公式: $$ r_n = \max_{1<=i<=n}(pi+r_{n-i}) $$ 动态规划就是不用递归,而是从底向上计算值,每次计算上面的值的时候,下面的值算好了,直接拿来用就行。因此咱们须要一个数组来记录每一个长度对应的最大收益。

const p = [0, 1, 5, 8, 9, 10, 17, 17, 20, 24, 30]; // 下标表示钢条长度,值表示对应价格

function cut_rod3(n) {
  let r = [0, 1];   // r数组记录每一个长度的最大收益
  
  for(let i = 2; i <=n; i++) {
    let max = p[i];
    for(let j = 1; j <= i; j++) {
      let sum = p[j] + r[i - j];
      
      if(sum > max) {
        max = sum;
      }
    }
    
    r[i] = max;
  }
  
  console.log(r);
  return r[n];
}

cut_rod3(9);  // 结果仍是返回 25

咱们还能够把r数组也打出来看下,这里面存的是每一个长度对应的最大收益:

r = [0, 1, 5, 8, 10, 13, 17, 18, 22, 25]

使用动态规划将递归的指数级复杂度降到了双重循环,即$O(n^2)$的复杂度。

输出最佳方案

上面的动态规划虽然计算出来最大值,可是咱们并非知道这个最大值对应的切割方案是什么,为了知道这个方案,咱们还须要一个数组来记录切割一次时左边的长度,而后在这个数组中回溯来找出切割方案。回溯的时候咱们先取目标值对应的左边长度,而后右边剩下的长度右继续去这个数组找最优方案对应的左边切割长度。假设咱们左边记录的数组是:

leftLength = [0, 1, 2, 3, 2, 2, 6, 1, 2, 3]

咱们要求长度为9的钢条的最佳切割方案:

1. 找到leftLength[9], 发现值为3,记录下3为一次切割
2. 左边切了3以后,右边还剩6,又去找leftLength[6],发现值为6,记录下6为一次切割长度
3. 又切了6以后,发现还剩0,切完了,结束循环;若是还剩有钢条继续按照这个方式切
4. 输出最佳长度为[3, 6]

改造代码以下:

const p = [0, 1, 5, 8, 9, 10, 17, 17, 20, 24, 30]; // 下标表示钢条长度,值表示对应价格

function cut_rod3(n) {
  let r = [0, 1];   // r数组记录每一个长度的最大收益
  let leftLength = [0, 1];  // 数组leftLength记录切割一次时左边的长度
  let solution = [];
  
  for(let i = 2; i <=n; i++) {
    let max = p[i];
    leftLength[i] = i;     // 初始化左边为整块不切
    for(let j = 1; j <= i; j++) {
      let sum = p[j] + r[i - j];
      
      if(sum > max) {
        max = sum;
        leftLength[i] = j;  // 每次找到大的值,记录左边的长度
      } 
    }
    
    r[i] = max;
  }
  
  // 回溯寻找最佳方案
  let tempN = n;
  while(tempN > 0) {
    let left = leftLength[tempN];
    solution.push(left);
    tempN = tempN - left;
  }
  
  console.log(leftLength);  // [0, 1, 2, 3, 2, 2, 6, 1, 2, 3]
  console.log(solution);    // [3, 6]
  console.log(r);           // [0, 1, 5, 8, 10, 13, 17, 18, 22, 25]
  return {max: r[n], solution: solution};
}

cut_rod3(9);  // {max: 25, solution: [3, 6]}

最长公共子序列

上叙问题也能够用暴力穷举来求解,先列举出X字符串全部的子串,假设他的长度为m,则总共有$2^m$种状况,由于对于X字符串中的每一个字符都有留着和不留两种状态,m个字符的全排列种类就是$2^m$种。那对应的Y字符串就有$2^n$种子串, n为Y的长度。而后再遍历找出最长的公共子序列,这个复杂度很是高,我这里就不写了。

咱们观察两个字符串,若是他们最后一个字符相同,则他们的LCS就是两个字符串都去调最后一个字符的LCS再加一,若是他们最后一个字符不相同,那他们的LCS就是X去掉最后一个字符与Y的LCS,或者是X与Y去掉最后一个字符的LCS,是他们两个中较长的那一个。写成数学公式就是:

看着这个公式,一个规模为(i, j)的问题转化为了规模为(i-1, j-1)的问题,这不就又能够用递归求解了吗?

递归方案

公式都有了,不废话,直接写代码:

function lcs(str1, str2) {
  let length1 = str1.length;
  let length2 = str2.length;
  
  if(length1 === 0 || length2 === 0) {
    return 0;
  }
  
  let shortStr1 = str1.slice(0, -1);
  let shortStr2 = str2.slice(0, -1);
  if(str1[length1 - 1] === str2[length2 -  1]){
    return lcs(shortStr1, shortStr2) + 1;
  } else {
    let lcsShort2 = lcs(str1, shortStr2);
    let lcsShort1 = lcs(shortStr1, str2);
    
    return lcsShort1 > lcsShort2 ? lcsShort1 : lcsShort2;
  }
}

let result = lcs('ABBCBDE', 'DBBCD');
console.log(result);   // 4

动态规划

递归虽然能实现咱们的需求,可是复杂度是在过高,长一点的字符串须要的时间是指数级增加的。咱们仍是要用动态规划来求解,根据咱们前面讲的动态规划原理,咱们须要从小的往大的算,每算出一个值都要记下来。由于c(i, j)里面有两个变量,咱们须要一个二维数组才能存下来。注意这个二维数组的行数是X的长度加一,列数是Y的长度加一,由于第一行和第一列表示X或者Y为空串的状况。代码以下:

function lcs2(str1, str2) {
  let length1 = str1.length;
  let length2 = str2.length;
  
  // 构建一个二维数组
  // i表示行号,对应length1 + 1
  // j表示列号, 对应length2 + 1
  // 第一行和第一列所有为0
  let result = [];
  for(let i = 0; i < length1 + 1; i++){
    result.push([]); //初始化每行为空数组
    for(let j = 0; j < length2 + 1; j++){
      if(i === 0) {
        result[i][j] = 0; // 第一行所有为0
      } else if(j === 0) {
        result[i][j] = 0; // 第一列所有为0
      } else if(str1[i - 1] === str2[j - 1]){
        // 最后一个字符相同
        result[i][j] = result[i - 1][j - 1] + 1;
      } else{
        // 最后一个字符不一样
        result[i][j] = result[i][j - 1] > result[i - 1][j] ? result[i][j - 1] : result[i - 1][j];
      }
    }
  }
  
  console.log(result);
  return result[length1][length2]
}

let result = lcs2('ABCBDAB', 'BDCABA');
console.log(result);   // 4

上面的result就是咱们构造出来的二维数组,对应的表格以下,每一格的值就是c(i, j),若是$X_i = Y_j$,则它的值就是他斜上方的值加一,若是$X_i \neq Y_i$,则它的值是上方或者左方较大的那一个。

输出最长公共子序列

要输出LCS,思路仍是跟前面切钢条的相似,把每一步操做都记录下来,而后再回溯。为了记录操做咱们须要一个跟result二维数组同样大的二维数组,每一个格子里面的值是当前值是从哪里来的,固然,第一行和第一列仍然是0。每一个格子的值要么从斜上方来,要么上方,要么左方,因此:

1. 咱们用1来表示当前值从斜上方来
2. 咱们用2表示当前值从左方来
3. 咱们用3表示当前值从上方来

看代码:

function lcs3(str1, str2) {
  let length1 = str1.length;
  let length2 = str2.length;
  
  // 构建一个二维数组
  // i表示行号,对应length1 + 1
  // j表示列号, 对应length2 + 1
  // 第一行和第一列所有为0
  let result = [];
  let comeFrom = [];   // 保存来历的数组
  for(let i = 0; i < length1 + 1; i++){
    result.push([]); //初始化每行为空数组
    comeFrom.push([]);
    for(let j = 0; j < length2 + 1; j++){
      if(i === 0) {
        result[i][j] = 0; // 第一行所有为0
        comeFrom[i][j] = 0;
      } else if(j === 0) {
        result[i][j] = 0; // 第一列所有为0
        comeFrom[i][j] = 0;
      } else if(str1[i - 1] === str2[j - 1]){
        // 最后一个字符相同
        result[i][j] = result[i - 1][j - 1] + 1;
        comeFrom[i][j] = 1;      // 值从斜上方来
      } else if(result[i][j - 1] > result[i - 1][j]){
        // 最后一个字符不一样,值是左边的大
        result[i][j] = result[i][j - 1];
        comeFrom[i][j] = 2;
      } else {
        // 最后一个字符不一样,值是上边的大
        result[i][j] = result[i - 1][j];
        comeFrom[i][j] = 3;
      }
    }
  }
  
  console.log(result);
  console.log(comeFrom);
  
  // 回溯comeFrom数组,找出LCS
  let pointerI = length1;
  let pointerJ = length2;
  let lcsArr = [];   // 一个数组保存LCS结果
  while(pointerI > 0 && pointerJ > 0) {
    console.log(pointerI, pointerJ);
    if(comeFrom[pointerI][pointerJ] === 1) {
      lcsArr.push(str1[pointerI - 1]);
      pointerI--;
      pointerJ--;
    } else if(comeFrom[pointerI][pointerJ] === 2) {
      pointerI--;
    } else if(comeFrom[pointerI][pointerJ] === 3) {
      pointerJ--;
    }
  }
  
  console.log(lcsArr);   // ["B", "A", "D", "B"]
  //如今lcsArr顺序是反的
  lcsArr = lcsArr.reverse();
  
  return {
    length: result[length1][length2], 
    lcs: lcsArr.join('')
  }
}

let result = lcs3('ABCBDAB', 'BDCABA');
console.log(result);   // {length: 4, lcs: "BDAB"}

最短编辑距离

这是leetcode上的一道题目,题目描述以下:

这道题目的思路跟前面最长公共子序列很是像,咱们一样假设第一个字符串是$X=(x_1, x_2 ... x_m)$,第二个字符串是$Y=(y_1, y_2 ... y_n)$。咱们要求解的目标为$r$, $r[i][j]$为长度为$i$的$X$和长度为$j$的$Y$的解。咱们一样从两个字符串的最后一个字符开始考虑:

  1. 若是他们最后一个字符是同样的,那最后一个字符就不须要编辑了,只须要知道他们前面一个字符的最短编辑距离就好了,写成公式就是:若是$Xi = Y_j$,$r[i][j] = r[i-1][j-1]$。
  2. 若是他们最后一个字符是不同的,那最后一个字符确定须要编辑一次才行。那最短编辑距离就是$X$去掉最后一个字符与$Y$的最短编辑距离,再加上最后一个字符的一次;或者是是$Y$去掉最后一个字符与$X$的最短编辑距离,再加上最后一个字符的一次,就看这两个数字哪一个小了。这里须要注意的是$X$去掉最后一个字符或者$Y$去掉最后一个字符,至关于在$Y$上进行插入和删除,可是除了插入和删除两个操做外,还有一个操做是替换,若是是替换操做,并不会改变两个字符串的长度,替换的时候,距离为$r[i][j]=r[i-1][j-1]+1$。最终是在这三种状况里面取最小值,写成数学公式就是:若是$Xi \neq Y_j$,$r[i][j] = \min(r[i-1][j], r[i][j-1],r[i-1][j-1]) + 1$。
  3. 最后就是若是$X$或者$Y$有任意一个是空字符串,那为了让他们同样,就往空的那个插入另外一个字符串就好了,最短距离就是另外一个字符串的长度。数学公式就是:若是$i=0$,$r[i][j] = j$;若是$j=0$,$r[i][j] = i$。

上面几种状况总结起来就是 $$ r[i][j]= \begin{cases} j, & \text{if}\ i=0 \ i, & \text{if}\ j=0 \ r[i-1][j-1], & \text{if}\ X_i=Y_j \ \min(r[i-1][j], r[i][j-1], r[i-1][j-1]) + 1, & \text{if} \ X_i\neq Y_j \end{cases} $$

递归方案

老规矩,有了递推公式,咱们先来写个递归:

const minDistance = function(str1, str2) {
    const length1 = str1.length;
    const length2 = str2.length;

    if(!length1) {
        return length2;
    }

    if(!length2) {
        return length1;
    }

    const shortStr1 = str1.slice(0, -1);
    const shortStr2 = str2.slice(0, -1); 

    const isLastEqual = str1[length1-1] === str2[length2-1];

    if(isLastEqual) {
        return minDistance(shortStr1, shortStr2);
    } else {
        const shortStr1Cal = minDistance(shortStr1, str2);
        const shortStr2Cal = minDistance(str1, shortStr2);
        const updateCal = minDistance(shortStr1, shortStr2);

        const minShort = shortStr1Cal <= shortStr2Cal ? shortStr1Cal : shortStr2Cal;
        const minDis = minShort <= updateCal ? minShort : updateCal;

        return minDis + 1;
    }
}; 

//测试一下
let result = minDistance('horse', 'ros');
console.log(result);  // 3

result = minDistance('intention', 'execution');
console.log(result);  // 5

动态规划

上面的递归方案提交到leetcode会直接超时,由于复杂度过高了,指数级的。仍是上咱们的动态规划方案吧,跟前面相似,须要一个二维数组来存放每次执行的结果。

const minDistance = function(str1, str2) {
    const length1 = str1.length;
    const length2 = str2.length;

    if(!length1) {
        return length2;
    }

    if(!length2) {
        return length1;
    }

    // i 为行,表示str1
    // j 为列,表示str2
    const r = [];
    for(let i = 0; i < length1 + 1; i++) {
        r.push([]);
        for(let j = 0; j < length2 + 1; j++) {
            if(i === 0) {
                r[i][j] = j;
            } else if (j === 0) {
                r[i][j] = i;
            } else if(str1[i - 1] === str2[j - 1]){ // 注意下标,i,j包括空字符串,长度会大1
                r[i][j] = r[i - 1][j - 1];
            } else {
                r[i][j] = Math.min(r[i - 1][j ], r[i][j - 1], r[i - 1][j - 1]) + 1;
            }
        }
    }

    return r[length1][length2];
};

//测试一下
let result = minDistance('horse', 'ros');
console.log(result);  // 3

result = minDistance('intention', 'execution');
console.log(result);  // 5

上述代码由于是双重循环,因此时间复杂度是$O(n^2)$。

总结

动态规划的关键点是要找出递推式,有了这个递推式咱们能够用递归求解,也能够用动态规划。用递归时间复杂度一般是指数级增加,因此咱们有了动态规划。动态规划的关键点是从小往大算,将每个计算记过的值都记录下来,这样咱们计算大的值的时候直接就取到前面计算过的值了。动态规划能够大大下降时间复杂度,可是增长了一个存计算结果的数据结构,空间复杂度会增长。这也算是一种用空间换时间的策略了。

原文出处:https://www.cnblogs.com/dennisj/p/12298123.html

相关文章
相关标签/搜索