LeetCode从低效到高效,点击

一、题目描述：

题目要求

假设你是一位很棒的家长，想要给你的孩子们一些小饼干。但是，每个孩子最多只能给一块饼干。对每个孩子 i，都有一个胃口值 g[i]，这是能让孩子们满足胃口的饼干的最小尺寸；并且每块饼干 j，都有一个尺寸 s[j] 。如果 s[j] >= g[i]，我们可以将这个饼干 j 分配给孩子 i ，这个孩子会得到满足。你的目标是尽可能满足越多数量的孩子，并输出这个最大数值。

示例

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

二、思路分析：

本题是一道匹配问题，解决思路是让刚刚满足孩子是最优解，为了优先方便满足最小的孩子的胃口，所以需要对这两个数组都进行排序，当一块饼干不能满足最小的胃口的孩子时，这个饼干是无效的。

三、AC 代码：

按照上面的思路很容易写出下面的暴力模拟代码：

// 44 ms 17.1 MB
int findContentChildren_v1(vector<int> &g, vector<int> &s)
{
    sort(g.begin(), g.end());
    sort(s.begin(), s.end());
    auto gi = g.begin();
    auto si = s.begin();
    int sum = 0;
    while (gi != g.end() && si != s.end())
    {
        cout << *gi << " " << *si << endl;
        if (*gi > *si)
        {
            // 如果不能满足孩子的胃口，饼干移动到下一个
            si++;
        }
        else
        {
            // 能够满足孩子的胃口，孩子、饼干后移，匹配数加1
            gi++;
            si++;
            sum++;
        }
    }
    return sum;
}

但是，上述代码还有优化空间，可以复用一些变量，最终优化成下面这样

// 32 ms 17.1 MB
int findContentChildren(vector<int> &g, vector<int> &s)
{
    sort(g.begin(), g.end());
    sort(s.begin(), s.end());
    auto gi = g.begin();
    auto si = s.begin();
    auto ge = g.end();
    auto se = s.end();
    int sum = 0;
    while (gi != ge && si != se)
    {
        if (*gi <= *si)
        {
            gi++;
            sum++;
        }
        si++;
    }
    return sum;
}

四、总结：

简单的贪心算法也可以通过节约变量,优化流程提高效率,注意对参数的复用

本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情