代码随想录算法训练营第二十九天 | 贪心算法 part03

134. 加油站

贪心策略：1. 如果每个加油站的剩余油量总和小于0，说明不存在可以绕一圈的情况
2. 在遍历加油站的过程中，当剩余油量和小于0时，说明此时的初始位置油量不能支持，因此更新初始位置到下一个加油站。

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int restSum = 0;
        int start = 0;
        int curSum = 0;

        for (int i = 0; i < gas.size(); ++i) {
            int rest = (gas[i] - cost[i]);
            restSum += rest;
            curSum += rest;
            if (curSum < 0) {
                curSum = 0;
                start = i + 1;
            }
        }
        if (restSum < 0)
            return -1;
        return start;
    }
};

135. 分发糖果

先分发左大于右的情况，再分发右大于左的情况。

class Solution {
public:
    int candy(vector<int>& ratings) {
        vector<int> candy(ratings.size(), 1);
        for (int i = 1; i < ratings.size(); ++i) {
            if (ratings[i] > ratings[i - 1]) {
                candy[i] = candy[i - 1] + 1;
            }
        }

        for (int i = ratings.size() - 2; i >= 0; --i) {
            if (ratings[i] > ratings[i + 1]) {
                candy[i] = max(candy[i], candy[i + 1] + 1);
            }
        }
        return accumulate(candy.begin(), candy.end(), 0);
    }
};

860.柠檬水找零

这一题感觉就是简单的分类讨论。

class Solution {
public:
    bool lemonadeChange(vector<int>& bills) {
        int five = 0;
        int ten = 0;
        for (int i : bills) {
            if (i == 5)
                five++;
            else if (i == 10) {
                if (five == 0)
                    return false;
                else {
                    five--;
                    ten++;
                }
            }
            else {
                if (ten > 0 && five > 0) {
                    ten--;
                    five--;
                }
                else if (five >= 3) {
                    five -= 3;
                }
                else 
                    return false;
            }
        }
        return true;
    }
};

406.根据身高重建队列

贪心策略是：首先根据身高进行降序排序，然后遍历数组，可以直接将数组根据前面有几个人作为下标插入新数组。

class Solution {
    static bool cmp(vector<int>& a, vector<int>& b) {
        if (a[0] == b[0])
            return a[1] < b[1];
        else
            return a[0] > b[0];
    }

public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), cmp);
        list<vector<int>> queue;
        for (vector<int> p : people) {
            int position = p[1];
            auto it = queue.begin();
            advance(it, position);
            queue.insert(it, p);
        }
        return vector<vector<int>>(queue.begin(), queue.end());
    }
};