贪心算法之JavaScript 解决分发饼干问题

110 阅读3分钟

日新计划455. 分发饼干

假设你是一位很棒的家长,想要给你的孩子们一些小饼干。但是,每个孩子最多只能给一块饼干。

对每个孩子 i,都有一个胃口值 g[i],这是能让孩子们满足胃口的饼干的最小尺寸;并且每块饼干 j,都有一个尺寸 s[j] 。如果 s[j] >= g[i],我们可以将这个饼干 j 分配给孩子 i ,这个孩子会得到满足。你的目标是尽可能满足越多数量的孩子,并输出这个最大数值。

  示例 1:

输入: g = [1,2,3], s = [1,1]
输出: 1
解释: 
你有三个孩子和两块小饼干,3个孩子的胃口值分别是:1,2,3。
虽然你有两块小饼干,由于他们的尺寸都是1,你只能让胃口值是1的孩子满足。
所以你应该输出1。
复制代码

示例 2:

输入: g = [1,2], s = [1,2,3]
输出: 2
解释: 
你有两个孩子和三块小饼干,2个孩子的胃口值分别是1,2。
你拥有的饼干数量和尺寸都足以让所有孩子满足。
所以你应该输出2.
复制代码

提示:

  • 1 <= g.length <= 3 * 10^4
  • 0 <= s.length <= 3 * 10^4
  • 1 <= g[i], s[j] <= 2^31 - 1

使用 JavaScript 也可以解决这个问题。下面是使用 JavaScript 实现贪心策略解决这个问题的代码:

function findContentChildren(g, s) {
    g.sort((a, b) => a - b); // 孩子的胃口值从小到大排序
    s.sort((a, b) => a - b); // 饼干的尺寸从小到大排序
    let i = 0, j = 0, res = 0;
    while (i < g.length && j < s.length) {
        if (s[j] >= g[i]) {
            // 如果饼干的尺寸能够满足孩子的胃口值,将这个饼干分配给孩子
            i++;
            res++;
        }
        j++;
    }
    return res;
}

console.log(
  "findContentChildren([1,2], [1,2,3])",
  findContentChildren([1, 2], [1, 2, 3])
); //findContentChildren([1,2], [1,2,3]) 2

在这个代码中,g 是一个数组,存储了每个孩子的胃口值;s 是一个数组,存储了每块饼干的尺寸。函数 findContentChildren 返回一个整数,表示能够被满足胃口的孩子的最大数量。

代码中首先对 gs 数组进行排序,然后使用 ij 分别表示当前孩子和当前饼干的下标。当当前饼干的尺寸能够满足当前孩子的胃口值时,将这个饼干分配给孩子并将 ires 都加 1。在每次循环中,j 都会加 1,即使当前饼干无法满足当前孩子的胃口,也需要尝试下一块饼干。

最后返回变量 res 的值,即为能够被满足胃口的孩子的最大数量。

function findContentChildren(g, s) {
    g.sort((a, b) => a - b);
    s.sort((a, b) => a - b);
    let i = 0;
    for (let j = 0; i < g.length && j < s.length; j++) {
        if (s[j] >= g[i]) i++;
    }
    return i;
}

console.log(
  "findContentChildren([1,2], [1,2,3])",
  findContentChildren([1, 2], [1, 2, 3])
);  //findContentChildren([1,2], [1,2,3]) 2

这个版本的代码使用了 for 循环代替了 while 循环,同时将变量 res 删除,并将其和 i 合并为一个变量。这个代码更加简洁,但是原理和之前的版本是一样的。

function findContentChildren(g, s) {
    g.sort((a, b) => a - b);
    s.sort((a, b) => a - b);
    return s.reduce((i, size) => {
        if (size >= g[i]) return i + 1;
        else return i;
    }, 0);
}
console.log(
  "findContentChildren([1,2], [1,2,3])",
  findContentChildren([1, 2], [1, 2, 3])
);  //findContentChildren([1,2], [1,2,3]) 2

这个版本的代码使用了 reduce() 函数来遍历饼干的尺寸。初始值为 0,表示还没有孩子的胃口被满足。对于每个饼干的尺寸,如果可以满足当前孩子的胃口,就将 i 加 1。最后,reduce() 函数返回的值就是被满足胃口的孩子的最大数量。这个版本的代码更加简洁,但是可能需要更多的时间去理解高阶函数的用法。

const findContentChildren = (g, s) =>
  (g.sort((a, b) => a - b), s.sort((a, b) => a - b), s.reduce((i, size) => size >= g[i] ? i + 1 : i, 0));
console.log(
  "findContentChildren([1,2], [1,2,3])",
  findContentChildren([1, 2], [1, 2, 3])
);  //findContentChildren([1,2], [1,2,3]) 2

这个版本的代码使用 ES6 的箭头函数来简写方法,原理是一样的。