434-贪心算法（134. 加油站、 135. 分发糖果、860. 柠檬水找零、406. 根据身高重建队列）

最新推荐文章于 2024-07-20 22:53:12 发布

liufeng2023

最新推荐文章于 2024-07-20 22:53:12 发布

阅读量96

点赞数

分类专栏：代码随想录刷题文章标签：贪心算法算法

本文链接：https://blog.csdn.net/Edward_LF/article/details/125507976

版权

代码随想录刷题专栏收录该内容

33 篇文章 1 订阅

订阅专栏

134. 加油站

在这里插入图片描述

1、暴力解法

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        for (int i = 0; i < cost.size(); i++)
        {
            int rest = gas[i] - cost[i];
            int index = (i + 1) % cost.size();

            while (rest > 0 && index != i)
            {
                rest += gas[index] - cost[index];
                index = (index + 1) % cost.size();
            }
            if (index == i && rest >= 0) return i;
        }
        return -1;
    }
};

2、贪心算法

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

        for (int i = 0; i < gas.size(); i++)
        {
            cur_sum += gas[i] - cost[i];
            total_sum += gas[i] - cost[i];

            if (cur_sum < 0)
            {
                start = i + 1;
                cur_sum = 0;
            }
        }

        if (total_sum < 0)  return -1;
        return start;
    }
};

在这里插入图片描述

135. 分发糖果

在这里插入图片描述

class Solution {
public:
    int candy(vector<int>& ratings) {
        
        vector<int> candyVec(ratings.size(), 1);

        for (int i = 1; i < ratings.size(); i++)
        {
            if (ratings[i] > ratings[i - 1])   candyVec[i] = candyVec[i - 1] + 1;
        }

        for (int i = ratings.size() - 2; i >= 0; i--)
        {
            if (ratings[i] > ratings[i + 1])   candyVec[i] = max(candyVec[i], candyVec[i + 1] + 1);
        }

        int res = 0;
        for (int c : candyVec) { res += c; }
        return res;
    }
};

在这里插入图片描述

860. 柠檬水找零

在这里插入图片描述

class Solution {
public:
    bool lemonadeChange(vector<int>& bills) {
        int five = 0, ten = 0, twenty = 0;

        for (int bill : bills)
        {
            if (bill == 5)   five++;
            if (bill == 10)
            {
                if (five <= 0)   return false;
                ten++;
                five--;
            }

            if (bill == 20)
            {
                if (five > 0 && ten > 0)
                {
                    five--;
                    ten--;
                    twenty++;
                }
                else if (five >= 3)
                {
                    five -= 3;
                    twenty++;
                }
                else
                {
                    return false;
                }
            }
        }
        reurn true;
    }
};

在这里插入图片描述

406. 根据身高重建队列

在这里插入图片描述

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), [](const vector<int>& a, const vector<int>& b)
            {
                if (a[0] == b[0])    return a[1] < b[1];
                else return a[0] > b[0];
            });

        vector < vector<int>> que;

        for (int i = 0; i < people.size(); i++)
        {
            int pos = people[i][1];
            que.insert(que.begin() + pos, people[i]);
        }
        return que;
    }
};