找零问题与动态规划

今天岩岩抛出了一道 code war 上的题目,大意如下:

一个函数接收两个参数,第一个参数是数字,第二个参数是数字数组,求数组里的数字加起来等于第一个参数的所有情况,可以无限次使用数组里的数字。
譬如 5, [1, 2, 5],总共有

  • 1 + 1 + 1 + 1 + 1 = 5
  • 1 + 1 + 1 + 2 = 5
  • 1 + 2 + 2 = 5
  • 5 = 5
    这样 4 种情况,所以返回 4。
    第一个参数为 0 的时候,返回 1。

后来我发现在 leet code 也有类似的题,是个找零问题,就是不同面值的硬币组合成一个数有多少种情况。还挺有意思的,我就做了一下,用了递归:

const change = function (sum, arr) {
    const nums = arr.sort((a, b) => a - b);
    return add(sum, nums);
};

const add = function (sum, nums) {
    const min = nums[0];
    if (sum === 0 || sum === min) return 1;

    let res = 0;
    if (nums.includes(sum)) res += 1;

    for (let i = 0; i < nums.length; i++) {
        const last = sum - nums[i];
        if (last >= nums[i]) res += add(last, nums.slice(i));
    }
    return res;
}

在 code war 上是成功提交了,但在 leet code 上却超时了,性能太差了。然后岩岩在 code war 的答案里看到了这样一段代码:

const countChange = (m, c) => {
  const a = [1].concat(Array(m).fill(0));
  for (let i = 0, l = c.length; i < l; i++) 
    for (let x = c[i], j = x; j <= m; j++) 
      a[j] += a[j - x];
  return a[m];
}

看得我俩十分懵逼,苦思半天还是懵逼。于是我上网搜到了相同解法的 C++ 版本:

using namespace std;

class Solution 
{
public:
    int change(int amount, vector<int>& coins) 
    {
        vector<int> dp(amount + 1, 0);
        dp[0] = 1;
        for (int coin : coins) 
        {
            for (int i = coin; i <= amount; ++i) 
            {
                dp[i] += dp[i - coin];
            }
        }
        return dp[amount];
    }
};

文章 说是用的动态规划(知道了这点很关键),虽然没具体解释,但这个版本的命名比上面那个 JS 版本实在是好懂太多了,dp 就是 dynamic programming 嘛,所以已经可以勉强推导他的思路:

dp 是一个长为 amount + 1 的表,依次用来记录组合成 0、1、2、3、、、amount 各有多少种情况,dp[0] 初始为 1,其他都初始为 0。

接下来循环硬币值,记录从当前硬币值到 amount 的所有组合情况。状态转移方程为:dp[i] += dp[i - coin]。什么意思呢?假设我们求 [1, 2, 5] 这三个面值组成 5 的情况,现在我先拿出一个 2,那我是不是只要再有一个 3 就可以得到 5 了,那我只要计算有多少种组合成 3 的情况就好了,即当 coin = 2 的时候,dp[5]new = dp[5]old + dp[3],以此类推。

for (int coin : coins) 
{
    for (int i = coin; i <= amount; ++i) 
    {
        dp[i] += dp[i - coin];
    }
}

以上这段代码的意思就是先计算我拿出 1 的时候,组合成 12345 的情况数,再计算当我拿出 2 的时候,组合成 2345 的情况数(加上拿出 1 时候的情况数),再计算拿出 5 的时候,组合成 5 的情况数(加上拿出 1 和拿出 2 时候的情况数),最后得出的 dp[5] 就是我们想要的结果。你可能会问我们要的是 dp[5],中间的 dp[1]dp[2]...dp[4] 有什么用,其实这就是动态规划的精髓,会把子问题的解记录(缓存)下来,因为这些子问题会在计算过程中多次用到,就不需要每次都计算了。

上述解法的大体思路其实和下面这个朴素递归是相似的,都是把问题分解为子问题进行求解,动态规划强就强在会缓存子问题的解避免重复计算从而提高效率。

const countChange = function(money, coins) {
  if (money < 0 || coins.length === 0)
    return 0
  else if (money === 0)
    return 1
  else
    return countChange(money - coins[0], coins) + countChange(money, coins.slice(1))
}

以后当你碰到一个问题,它可以分解为多个子问题,并且子问题有重叠时,就用动态规划吧。

啊,我真是太菜了……一个动态规划的题搞了半天……

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

推荐阅读更多精彩内容

  • 动态规划(Dynamic Programming) 本文包括: 动态规划定义 状态转移方程 动态规划算法步骤 最长...
    廖少少阅读 3,273评论 0 18
  • 0. 动态规划分析 0.1 动态规划、递归和贪心算法的区别 动态规划就是利用分治思想和解决冗余的办法来处理问题,所...
    dreamsfuture阅读 7,410评论 2 6
  • 算法思想贪心思想双指针排序快速选择堆排序桶排序荷兰国旗问题二分查找搜索BFSDFSBacktracking分治动态...
    第六象限阅读 3,071评论 0 0
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,963评论 25 707
  • 张文质老师说:“美好的人总会相遇。” 是的,我确信。 1 2004年9月,我正式开始了教师生涯。 在见习期里,我遇...
    陈蕾FZ阅读 880评论 0 50