JS动态规划算法--01背包问题

动态规划

动态规划算法是通过拆分问题,定义问题状态和状态之间的关系,使得问题能够以递推(或者说分治)的方式去解决。

01 背包问题

有 N 件物品和一个容量为 V 的背包。第 i 件物品的体积是 c[i],价值是 w[i]。求解将哪些物品装入背包可使价值总和最大(每一种物品只能放一次)

分析解答

也不多扯皮,直入正题。

先看一个简单的问题,斐波那契数列:

dynFib(n) {
  let arr = new Array(n).fill(0);
  arr[0] = 1;
  arr[1] = 1;
  for (let i = 2; i < n; i++) {
    arr[i] = arr[i - 1] + arr[i - 2];
  }
  // 返回最后一个
  return arr[n - 1];
}

该问题就是一个简单的动态规划问题,关键点在于找到状态关系: f(n) = f(n-1) + f(n-2);

回到背包问题,可以参考上面斐波那契数列的递推关系,

假设现在背包容量为 10,有一个物品数组:

const product = [
  { weight: 2, value: 3},
  { weight: 2, value: 6 },
  { weight: 6, value: 5 },
  { weight: 5, value: 4 },
  { weight: 4, value: 6 },
];

如果背包容量为 0,那么什么也放不下

如果背包容量为 1,还是什么也放不下

如果背包容量为 2,有两个物品都为质量都为 2,该放哪一个呢?

这时需要做一个决策,先放物品 1,此时价值为 3;此时比较 上一个决策的价值 - 物品 1 的价值 + 放物品 2 的价值 和物品 1 的价值比,取大的当做该容量下的最优解

用状态方程描述为: f[i][j]= Math.max(f[i-1][j],f[i-1][j]-c[i]]+w[i]); 其中 i 为第 i 件物品(0 开始), j 为当前背包容量

如果背包容量为 3,.......

......

直到背包容量为 10,得到此时的最优解


初始化一个二维矩阵来记录背包和价值的关系:

this.result = [];

const row = this.product.length;
const col = this.capacity;
for (let i = 0; i < row; i++) {
  this.result[i] = new Array(col + 1).fill(0);
}

我们会发现,每一个状态都依赖于上一个状态,第一个物品需要比较没有物品的状态做比较,我们需要对 i=0 时做特殊处理,还有一个更巧妙的方法,初始化一个 -1 的状态:

this.result[-1] = new Array(col + 1).fill(0);

如下图,我们发现类似于填表的方式,右下角即为背包最大容量时的价值;

观察下表可以发现,该矩阵记录了每一个容量即状态的最优解,比如容量 7 时的最优解,我们可以找到 7 那一列最下面即 12;

主函数部分,利用状态方程进行最优决策

calculate() {
  const row = this.product.length;
  const col = this.capacity;
  let product;
  for (let i = 0; i < row; i++) {
    for (let j = 0; j < col + 1; j++) {
      product = this.product[i];
      if (j < product.weight) {
        this.result[i][j] = this.result[i - 1][j];
      } else {
        this.result[i][j] = Math.max(
          this.result[i - 1][j],
          this.result[i - 1][j - product.weight] + product.value
        );
      }
    }
  }
  delete this.result[-1];
  return this.result[row - 1][col];
}

动态规划解决问题主要是两个:

  • 拆分、划分问题到某个颗度时,我们可以很轻易的做出决策
  • 确定状态方程,比如上面的: f[i][j]= Math.max(f[i-1][j],f[i-1][j]-c[i]]+w[i])

源码

源码


END

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342