24暑假算法刷题 | Day29 | 贪心算法 III | LeetCode 134. 加油站，135. 分发糖果，860. 柠檬水找零，406. 根据身高重建队列

---

134. 加油站

点此跳转题目链接

题目描述

在一条环路上有 n 个加油站，其中第 i 个加油站有汽油 gas[i] 升。

你有一辆油箱容量无限的的汽车，从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发，开始时油箱为空。

给定两个整数数组 gas 和 cost ，如果你可以按顺序绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1 。如果存在解，则 保证 它是 唯一 的。

示例 1:

输入: gas = [1,2,3,4,5], cost = [3,4,5,1,2]
输出: 3
解释:
从 3 号加油站(索引为 3 处)出发，可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
开往 4 号加油站，此时油箱有 4 - 1 + 5 = 8 升汽油
开往 0 号加油站，此时油箱有 8 - 2 + 1 = 7 升汽油
开往 1 号加油站，此时油箱有 7 - 3 + 2 = 6 升汽油
开往 2 号加油站，此时油箱有 6 - 4 + 3 = 5 升汽油
开往 3 号加油站，你需要消耗 5 升汽油，正好足够你返回到 3 号加油站。
因此，3 可为起始索引。

示例 2:

输入: gas = [2,3,4], cost = [3,4,3]
输出: -1
解释:
你不能从 0 号或 1 号加油站出发，因为没有足够的汽油可以让你行驶到下一个加油站。
我们从 2 号加油站出发，可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油
开往 0 号加油站，此时油箱有 4 - 3 + 2 = 3 升汽油
开往 1 号加油站，此时油箱有 3 - 3 + 3 = 3 升汽油
你无法返回 2 号加油站，因为返程需要消耗 4 升汽油，但是你的油箱只有 3 升汽油。
因此，无论怎样，你都不可能绕环路行驶一周。

提示:

• gas.length == n
• cost.length == n
• 1 <= n <= 105
• 0 <= gas[i], cost[i] <= 104

题解

贪心算法解决。直觉上，容易考虑一种暴力方法：

依次以每个加油站为起点，每次将当前剩余油量加上 gas[i] 、再减去 cost[i] ，就得到了新的剩余油量；如果直到循环一圈回到起点加油站时，剩余油量每次都不为负，则说明有解、返回该起点加油站编号；否则，返回-1。

其实这已经比较符合贪心算法的思维了（局部最优推整体最优）。我们可以进一步简化：

如果从加油站 i 出发，到加油站 j 时，剩余油量第一次不足，则说明从 i 到 j 内的任意加油站出发，到 j 时剩余油量都会不足，于是类似地考虑以 j + 1 为起点的情况 …由于题目说了，要么有唯一解要么无解，我们这样遍历一次之后就能找到目标起点。

上面的结论证明起来也并不复杂：
$$\begin{array}{l}上述条件下，假设从[i,j]中某个加油站k出发，到j时剩余油量充足(G_{kj}\ge 0)\\ 由于从i出发，到j时剩余油量第一次不足(G_{ij}<0)\\ 且G_{ij}=G_{ik}+G_{kj}\\ 所以G_{ik}<0，即从i出发，到k时剩余油量就不足了，矛盾\\ 所以从[i,j]中任意加油站出发，到j时，剩余油量都会不足\end{array}$$
代码（C++）

int canCompleteCircuit(vector<int> &gas, vector<int> &cost)
{
    int curGasLeft = 0;   // 当前剩余的油量
    int totalGasLeft = 0; // 总剩余油量（= 总gas - 总cost）
    int start = 0;
    for (int i = 0; i < gas.size(); ++i)
    {
        curGasLeft += gas[i] - cost[i];
        totalGasLeft += gas[i] - cost[i];
        if (curGasLeft < 0)
        {
            start = i + 1; // 尝试新的起点
            curGasLeft = 0;
        }
    }
    if (totalGasLeft < 0)
        return -1;
    return start;
}

---

135. 分发糖果

点此跳转题目链接

题目描述

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

• 每个孩子至少分配到 1 个糖果。
• 相邻两个孩子评分更高的孩子会获得更多的糖果。

请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

示例 1：

输入：ratings = [1,0,2]
输出：5
解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。

示例 2：

输入：ratings = [1,2,2]
输出：4
解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
     第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。

提示：

• n == ratings.length
• 1 <= n <= 2 * 104
• 0 <= ratings[i] <= 2 * 104

题解

贪心算法解决。首先很容易想到贪心的初始状态：每个孩子先各分配一个糖果（题目说的最小值）。然后，如果相邻孩子 ratings 值更高，则分配给ta的糖果（只）增加1个。

由于对于非边界的任意位置，都需要考虑其左边和右边的两个 ratings 值，一开始容易陷入“两者兼顾”的思维泥潭，反而“顾此失彼”。其实我们可以通过两次遍历就解决这个问题：

• 先从左向右遍历： 如果 ratings[i] 大于其左邻值 ratings[i - 1] ，则给第 i 个孩子的糖果在第 i - 1 个孩子的基础上增加1颗；否则，保持只给一颗。
• 再从右向左遍历： 如果 ratings[i] 大于其右邻值 ratings[i + 1] ，则给第 i 个孩子的糖果在第 i + 1 个孩子的基础上增加1颗；否则，保持原状。

🤔 ​为什么不能两次都从左到右（或从右向左）遍历？

以都从左到右遍历为例，那么第二次遍历（判断右邻值情况）时， i 位置的值根据 i + 1 决定；然后（从左到右遍历）到 i + 1 时，该位置的值又可能由于 i + 2 的情况而改变，此时原来那个 i 位置的值又不得不根据最新的这个 i + 1 的情况而改变… 😅

进一步解析可参见 代码随想录-135

代码（C++）

int candy(vector<int> &ratings)
{
    int count = ratings.size();           // 总糖果数
    vector<int> child(ratings.size(), 1); // 分配给每个孩子的糖果数
    // 从左向右遍历，检查：当前比左边大的情况
    for (int i = 1; i < ratings.size(); ++i) {
        if (ratings[i] > ratings[i - 1]) {
            count += child[i - 1];
            child[i] += child[i - 1];
        }
    }
    // 从右向左遍历，检查：当前比右边大的情况
    for (int i = ratings.size() - 2; i >= 0; --i) {
        if (ratings[i] > ratings[i + 1] && child[i] <= child[i + 1]) {
            count += (child[i + 1] + 1 - child[i]);
            child[i] = child[i + 1] + 1;
        }
    }
    return count;
}

---

860. 柠檬水找零

点此跳转题目链接

题目描述

在柠檬水摊上，每一杯柠檬水的售价为 5 美元。顾客排队购买你的产品，（按账单 bills 支付的顺序）一次购买一杯。

每位顾客只买一杯柠檬水，然后向你付 5 美元、10 美元或 20 美元。你必须给每个顾客正确找零，也就是说净交易是每位顾客向你支付 5 美元。

注意，一开始你手头没有任何零钱。

给你一个整数数组 bills ，其中 bills[i] 是第 i 位顾客付的账。如果你能给每位顾客正确找零，返回 true ，否则返回 false 。

示例 1：

输入：bills = [5,5,5,10,20]
输出：true
解释：
前 3 位顾客那里，我们按顺序收取 3 张 5 美元的钞票。
第 4 位顾客那里，我们收取一张 10 美元的钞票，并返还 5 美元。
第 5 位顾客那里，我们找还一张 10 美元的钞票和一张 5 美元的钞票。
由于所有客户都得到了正确的找零，所以我们输出 true。

示例 2：

输入：bills = [5,5,10,10,20]
输出：false
解释：
前 2 位顾客那里，我们按顺序收取 2 张 5 美元的钞票。
对于接下来的 2 位顾客，我们收取一张 10 美元的钞票，然后返还 5 美元。
对于最后一位顾客，我们无法退回 15 美元，因为我们现在只有两张 10 美元的钞票。
由于不是每位顾客都得到了正确的找零，所以答案是 false。

提示：

• 1 <= bills.length <= 105
• bills[i] 不是 5 就是 10 或是 20

题解

非常简单的贪心算法，直接按生活常识的思路解决：每次尽量把大面值的钱找给顾客即可（小面值总是能找开的，贪心地尽量留着）。

代码（C++）

bool lemonadeChange(vector<int> &bills)
{
    // 贪心：尽量先把大面值的找出去
    int cash5 = 0, cash10 = 0;
    for (int bill : bills)
    {
        switch (bill)
        {
        case 5:
            cash5++;
            break;
        case 10:
            if (!cash5)
                return false;
            cash5--;
            cash10++;
            break;
        case 20:
            if (cash10 && cash5)
                cash10--, cash5--;
            else if (cash5 >= 3)
                cash5 -= 3;
            else
                return false;
            break;
        default:
            break;
        }
    }
    return true;
}

Go

func lemonadeChange(bills []int) bool {
	cash5, cash10 := 0, 0
	for _, bill := range bills {
		switch bill {
		case 5:
			cash5++
		case 10:
			if cash5 == 0 {
				return false
			}
			cash5--
			cash10++
		case 20:
			if cash10 > 0 && cash5 > 0 {
				cash10--
				cash5--
			} else if cash5 >= 3 {
				cash5 -= 3
			} else {
				return false
			}
		}
	}
	return true
}

---

406. 根据身高重建队列

点此跳转题目链接

题目描述

假设有打乱顺序的一群人站成一个队列，数组 people 表示队列中一些人的属性（不一定按顺序）。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ，前面 正好 有 ki 个身高大于或等于 hi 的人。

请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ，其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性（queue[0] 是排在队列前面的人）。

示例 1：

输入：people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出：[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释：
编号为 0 的人身高为 5 ，没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ，没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ，有 2 个身高更高或者相同的人排在他前面，即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ，有 1 个身高更高或者相同的人排在他前面，即编号为 1 的人。
编号为 4 的人身高为 4 ，有 4 个身高更高或者相同的人排在他前面，即编号为 0、1、2、3 的人。
编号为 5 的人身高为 7 ，有 1 个身高更高或者相同的人排在他前面，即编号为 1 的人。
因此 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。

示例 2：

输入：people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]]
输出：[[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]

提示：

• 1 <= people.length <= 2000
• 0 <= hi <= 106
• 0 <= ki < people.length
• 题目数据确保队列可以被重建

题解

贪心算法解决。核心思想和 135. 分发糖果 有异曲同工之妙，都是通过“两边分别贪心”避免“顾此失彼”的问题：

• 先贪心身高：将 people[i] = [hi, ki] 按照身高 hi 从高到矮的顺序排序（身高相同则 ki 更小的在前）——这种排法非常符合直觉。
• 再贪心前面比自己高（或一样高）的人数：再按照 ki ，将数据重新插入最新的队列。

借用 代码随想录 中的图展示一下：

简单模拟一下这个过程，可以发现：

每次插入新的数据时，总是在（ hi ，即身高）比它更高的数据之前位置插入，所以那些更高的数据不会受影响（每次插入的数据都更矮，不会影响 ki 值 。

👏 👏 👏 ​

代码（C++）

vector<vector<int>> reconstructQueue(vector<vector<int>> &people)
{
    // 贪心算法：先按身高排，再按前面更高人数排
    vector<vector<int>> queue;
    auto cmp = [](const vector<int> &a, const vector<int> &b) {
        if (a[0] != b[0])
            return a[0] > b[0];
        return a[1] < b[1];
    };
    sort(people.begin(), people.end(), cmp);
    for (const auto &info : people) 
        queue.insert(queue.begin() + info[1], info);
    return queue;
}