代码随想录-贪心算法①|ACM模式

目录

引言:

455. 分发饼干

题目描述:

输入输出示例:

思路和想法:

376. 摆动序列

题目描述:

输入输出示例:

思路和想法:

53. 最大子数组和

题目描述:

输入输出示例:

思路和想法:

122. 买卖股票的最佳时机 II

题目描述:

输入输出示例:

思路和想法:

55. 跳跃游戏

题目描述:

输入输入描述:

思路和想法:

45. 跳跃游戏 II

题目描述:

输入输出示例:

思路和想法:

1005. K 次取反后最大化的数组和

题目描述:

输入输出描述:

思路和想法:

134. 加油站

题目描述:

输入输出描述:

思路和想法:

135. 分发糖果

题目描述:

输入输出描述:

思路和想法:


引言:

贪心的本质是选择每一阶段的局部最优,从而达到全局最优。

贪心算法唯一的难点就是如何通过局部最优,推出整体最优。如果可以靠自己手动模拟,则可以试一试贪心策略,如果不可行,可能需要动态规划。

刷题时,手动模拟一下感觉可以局部最优推出整体最优,并且想不到反例,就可以试一试贪心。

贪心算法的一般步骤分为四步:

  • 将问题分解为若干个子问题
  • 找出适合的贪心策略
  • 求解每一个子问题的最优解
  • 将局部最优解堆叠成全局最优解

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 * 104
  • 0 <= s.length <= 3 * 104
  • 1 <= g[i], s[j] <= 231 - 1

思路和想法:

这里我先想着对胃口和饼干进行排序,在一个胃口的同时,遍历饼干数据,看是否满足。一句话概括,小饼干优先小胃口。

#include <bits/stdc++.h>
using namespace std;

int findContentChildren(vector<int>& g, vector<int>& s) {
    sort(g.begin(),g.end());
    sort(s.begin(),s.end());
    int result = 0;
    for (int i = 0, j = 0; i < g.size() && j < s.size(); i++,j++)
        {
            while(j < s.size() && g[i] > s[j]){
                j++;
            }
            if(j < s.size()){
                result++;
            }
        }
    return result;

}
int main(void){
    int num;
    vector<int> g,s;
    while(cin >> num){
        g.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }
    while(cin >> num){
        s.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }

    int result;
    result = findContentChildren(g,s);
    cout << result << endl;
    return 0;
}
/*
示例1:
1 2 3
1 1
示例2:
1 2
1 2 3
 * */

376. 摆动序列

题目描述:

如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为 摆动序列 。第一个差(如果存在的话)可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。

  • 例如, [1, 7, 4, 9, 2, 5] 是一个 摆动序列 ,因为差值 (6, -3, 5, -7, 3) 是正负交替出现的。
  • 相反,[1, 4, 7, 2, 5] 和 [1, 7, 4, 5, 5] 不是摆动序列,第一个序列是因为它的前两个差值都是正数,第二个序列是因为它的最后一个差值为零。

子序列 可以通过从原始序列中删除一些(也可以不删除)元素来获得,剩下的元素保持其原始顺序。

给你一个整数数组 nums ,返回 nums 中作为 摆动序列 最长子序列的长度

输入输出示例:

示例 1:

输入:nums = [1,7,4,9,2,5] 输出:6

解释:整个序列均为摆动序列,各元素之间的差值为 (6, -3, 5, -7, 3) 。

示例 2:

输入:nums = [1,17,5,10,13,15,10,5,16,8] 输出:7

解释:这个序列包含几个长度为 7 摆动序列。 其中一个是 [1, 17, 10, 13, 10, 16, 8] ,各元素之间的差值为 (16, -7, 3, -3, 6, -8) 。

示例 3:

输入:nums = [1,2,3,4,5,6,7,8,9] 输出:2

提示:

  • 1 <= nums.length <= 1000
  • 0 <= nums[i] <= 1000

思路和想法:

这个序列是不能进行排序的,是要找到作为摆动序列的最长子序列。核心思路是找到序列的峰和谷底。这里有三个特殊情况,分别如下:

  • 上下坡中有平坡
  • 数组首尾两端
  • 单调坡中有平坡---在有坡度变化时,才更新preDiff
#include <bits/stdc++.h>
using namespace std;

int wiggleMaxLength(vector<int>& nums) {
    if(nums.size() <= 1) return nums.size();
    int preDiff = 0;        //前一对差值
    int curDiff = 0;        //当前一对差值
    int result = 1;         //默认从1开始
    //三种特殊情况
    //情况一:上下坡中右平坡
    //情况二:数组首尾两端
    //情况三:单调坡中有平坡
    for(int i = 0; i < nums.size() - 1; i++){
        curDiff = nums[i + 1] -nums[i];
        if(preDiff <= 0 && curDiff > 0 || preDiff >= 0 && curDiff < 0){
            result++;
            preDiff = curDiff;
        }
    }
    return result;
}

int main(void){
    int num;
    vector<int> nums;
    while (cin >> num){
        nums.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }

    int result;
    result = wiggleMaxLength(nums);
    cout << result << endl;
    return 0;
}
/*
示例1:
1 7 4 9 2 5
示例2:
1 17 5 10 13 15 10 5 16 8
示例3:
1 2 3 4 5 6 7 8 9
 * */

53. 最大子数组和

题目描述:

给你一个整数数组 nums ,请你找出一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。

子数组 是数组中的一个连续部分。

输入输出示例:

示例 1:

输入:nums = [-2,1,-3,4,-1,2,1,-5,4] 输出:6

解释:连续子数组 [4,-1,2,1] 的和最大,为 6 。

示例 2:

输入:nums = [1] 输出:1

示例 3:

输入:nums = [5,4,-1,7,8] 输出:23

提示:

  • 1 <= nums.length <= 105
  • -104 <= nums[i] <= 104

思路和想法:

局部最优:当前“连续和”为负数的时候立刻放弃,从下一个元素重新计算“连续和”,因为负数加上下一个元素 “连续和”只会越来越小。

全局最优:选取最大“连续和”

#include <bits/stdc++.h>
using namespace std;

int maxSubArray(vector<int>& nums) {
    int result = INT32_MIN;
    int count = 0;
    for(int i = 0; i < nums.size(); i++){
        count += nums[i];
        if(count > result){
            result = count;
        }
        if(count <= 0) count = 0;
    }
    return result;
}

int main(void){
    int num;
    vector<int> nums;
    while (cin >> num){
        nums.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }

    int result;
    result = maxSubArray(nums);
    cout << result << endl;
    return 0;
}
/*
示例1:
-2 1 -3 4 -1 2 1 -5 4
示例2:
1
示例3:
5 4 -1 7 8
 * */

122. 买卖股票的最佳时机 II

题目描述:

给你一个整数数组 prices ,其中 prices[i] 表示某支股票第 i 天的价格。

在每一天,你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买,然后在 同一天 出售。

返回 你能获得的 最大 利润

输入输出示例:

示例 1:

输入:prices = [7,1,5,3,6,4] 输出:7

解释:在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4 。 随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6 - 3 = 3 。 总利润为 4 + 3 = 7 。

示例 2:

输入:prices = [1,2,3,4,5] 输出:4

解释:在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4 。 总利润为 4 。

示例 3:

输入:prices = [7,6,4,3,1] 输出:0

解释:在这种情况下, 交易无法获得正利润,所以不参与交易可以获得最大利润,最大利润为 0 。

提示:

  • 1 <= prices.length <= 3 * 104
  • 0 <= prices[i] <= 104

思路和想法:

这道题局部最优只要找到上升沿即可,将上升沿相近数值的差加和起来即可。

#include <bits/stdc++.h>
using namespace std;

int maxProfit(vector<int>& prices) {
    int curDiff = 0;
    int result  = 0;
    for (int i = 0; i < prices.size() - 1; i++)
        {
            curDiff = prices[i + 1] - prices[i];
            if(curDiff > 0){
                result += curDiff;
            }
        }
    return result;
}
int main(void){
    int num;
    vector<int> nums;
    while (cin >> num){
        nums.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }

    int result;
    result = maxProfit(nums);
    cout << result << endl;
    return 0;
}
/*
示例1:
7 1 5 3 6 4
示例2:
1 2 3 4 5
示例3:
7 6 4 3 1
 * */

55. 跳跃游戏

题目描述:

给你一个非负整数数组 nums ,你最初位于数组的 第一个下标 。数组中的每个元素代表你在该位置可以跳跃的最大长度。

判断你是否能够到达最后一个下标,如果可以,返回 true ;否则,返回 false 。

输入输入描述:

示例 1:

输入:nums = [2,3,1,1,4] 输出:true

解释:可以先跳 1 步,从下标 0 到达下标 1, 然后再从下标 1 跳 3 步到达最后一个下标。

示例 2:

输入:nums = [3,2,1,0,4] 输出:false

解释:无论怎样,总会到达下标为 3 的位置。但该下标的最大跳跃长度是 0 , 所以永远不可能到达最后一个下标。

提示:

  • 1 <= nums.length <= 104
  • 0 <= nums[i] <= 105

思路和想法:

问题核心在于target的更新以及找到能到target的最远元素---局部最优。循环终止条件:它第一个元素能到target,这时返回true;跑完循环,未找到元素能到target,则返回false。

注:一个特殊情况需注意,就是只有一个元素时,返回true。

#include <bits/stdc++.h>
using namespace std;

bool canJump(vector<int>& nums) {
    //问题核心在于target的更新以及找到能到target的元素
    if(nums.size() <= 1){
        return true;
    }
    int target = nums.size() - 1;
    for (int i = 0; i <= target - 1; i++)
        {
            int h = i + nums[i];
            //一次遍历,能找到相应的元素能跳到目标
            if(i == 0 && h >= target){
                return true;
            }

            if(h >= target){
                target = i;
                i = -1;
            }
        }
    //一次遍历结束,还未找到元素能跳到目标,则return false
    return false;
}

int main(void){
    int num;
    vector<int> nums;
    while (cin >> num){
        nums.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }

    bool result;
    result = canJump(nums);
    if(result == 1){
        cout << "true" << endl;
    }else {
        cout << "false" << endl;
    }
    return 0;
}
/*
示例1:
2 3 1 1 4
示例2:
3 2 1 0 4
 * */

45. 跳跃游戏 II

题目描述:

给定一个长度为 n 的 0 索引整数数组 nums。初始位置为 nums[0]。

每个元素 nums[i] 表示从索引 i 向前跳转的最大长度。换句话说,如果你在 nums[i] 处,你可以跳转到任意 nums[i + j] 处:

  • 0 <= j <= nums[i]
  • i + j < n

返回到达 nums[n - 1] 的最小跳跃次数。生成的测试用例可以到达 nums[n - 1]。

输入输出示例:

示例 1:

输入: nums = [2,3,1,1,4] 输出: 2

解释: 跳到最后一个位置的最小跳跃数是 2。 从下标为 0 跳到下标为 1 的位置,跳 1 步,然后跳 3 步到达数组的最后一个位置。

示例 2:

输入: nums = [2,3,0,1,4] 输出: 2

提示:

  • 1 <= nums.length <= 104
  • 0 <= nums[i] <= 1000
  • 题目保证可以到达 nums[n-1]

思路和想法:

这道题在跳跃游戏那道题的基础上需要再记录一下target更新次数---对应最小跳跃数。

注:这里的测试用例是一定能达到最后一个位置。

#include <bits/stdc++.h>
using namespace std;

int jump(vector<int>& nums) {
    //问题核心在于target的更新以及target更新了几次
    if(nums.size() <= 1){          //这时无需跳跃,跳跃次数为零
        return 0;
    }
    int target = nums.size() - 1;  //目标位置
    int result = 0;                 //记录target更新次数
    for (int i = 0; i <= target - 1; i++)
        {
            int h = i + nums[i];
            if(h >= target){
                result++;
                target = i;
                i = -1;
            }
        }
    return result;
}

int main(void){
    int num;
    vector<int> nums;
    while (cin >> num){
        nums.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }

    int result;
    result = jump(nums);
    cout << result << endl;
    return 0;
}
/*
示例1:
2 3 1 1 4
示例2:
2 3 0 1 4
 * */

1005. K 次取反后最大化的数组和

题目描述:

给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:

  • 选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。

重复这个过程恰好 k 次。可以多次选择同一个下标 i 。

以这种方式修改数组后,返回数组 可能的最大和

输入输出描述:

示例 1:

输入:nums = [4,2,3], k = 1 输出:5

解释:选择下标 1 ,nums 变为 [4,-2,3] 。

示例 2:

输入:nums = [3,-1,0,2], k = 3 输出:6

解释:选择下标 (1, 2, 2) ,nums 变为 [3,1,0,2] 。

示例 3:

输入:nums = [2,-3,-1,5,-4], k = 2 输出:13

解释:选择下标 (1, 4) ,nums 变为 [2,3,-1,5,4] 。

提示:

  • 1 <= nums.length <= 104
  • -100 <= nums[i] <= 100
  • 1 <= k <= 104

思路和想法:

这道题可以用两次贪心算法,来解决问题。第一次局部最优:将绝对值大的负数先翻转一次,k--,能推出全局最优-求和最大。

这时k值还是大于0,则进行第二次贪心算法:将绝对值最小的数进行翻转,这时对数组求和影响最小,能推出全局最优-求和最大。

#include <bits/stdc++.h>
using namespace std;

static bool cmp(int a, int b){
    return abs(a) > abs(b);
}

int largestSumAfterKNegations(vector<int>& nums, int k) {
    sort(nums.begin(), nums.end(), cmp);  //cmp排序的方式,绝对值排序,从大到小排序
    for(int i = 0; i < nums.size(); i++){   //第一次贪心
        if(nums[i] < 0 && k > 0){
            nums[i] *= -1;
            k--;
        }
    }
    if(k % 2 == 1) nums[nums.size() - 1] *= -1;   //第二次贪心
    return accumulate(nums.begin(),nums.end(),0);  //返回数组求和
}

int main(void){
    int num;
    vector<int> nums;
    while (cin >> num){
        nums.push_back(num);
        if(::getchar() == '\n'){
            break;
        }
    }
    int k;
    cin >> k;


    int result;
    result = largestSumAfterKNegations(nums, k);
    cout << result << endl;
    return 0;
}
/*
示例1:
4 2 3
1
示例2:
3 -1 0 2
3
示例3:
2 -3 -1 5 -4
2
 * */

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

思路和想法:

当前剩余rest[j](gas[i] - cost[i])和当前和curSum一旦小于0,起始位置至少要是j+1,因为从j开始一定不行。

全局最优:找到可以跑一圈的起始位置。

#include <bits/stdc++.h>
using namespace std;

int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
    int gasC = accumulate(gas.begin(), gas.end(), 0);
    int costC = accumulate(cost.begin(), cost.end(), 0);
    if(gasC < costC) return -1;

    int curSum = 0;
    int start = 0;  //起始位置
    for (int i = 0; i < gas.size(); i++)
        {
            curSum += gas[i] - cost[i];
            if(curSum < 0){
                start = i + 1;
                curSum = 0;
            }
        }
    return start;

}

int main(void){
    int num;
    vector<int> gas;
    while (cin >> num) {
        gas.push_back(num);
        if (::getchar() == '\n') {
            break;
        }
    }
    vector<int> cost;
    while (cin >> num) {
        cost.push_back(num);
        if (::getchar() == '\n') {
            break;
        }
    }


    int result;
    result = canCompleteCircuit(gas, cost);
    cout << result << endl;
    return 0;
}
/*
示例1:
1 2 3 4 5
3 4 5 1 2
示例2:
2 3 4
3 4 3
 * */

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

思路和想法:

本题没法两边都考虑,需要确定一边之后,再确定另一边;本题采用两次贪心的策略:

  • 一次是从左到右遍历,只比较右边孩子评分比左边大的情况。
  • 一次是从右到左遍历,只比较左边孩子评分比右边大的情况。

以第一次贪心为例,局部最优:只要右边评分比左边大,右边的孩子就多一个糖果,全局最优:相邻的孩子中,评分高的右孩子获得比左边孩子更多的糖果

#include <bits/stdc++.h>
using namespace std;

int candy(vector<int>& ratings) {
    vector<int> candy(ratings.size(),1);
    //从前向后,比左,右孩子大于左孩子
    for (int i = 0; i < ratings.size() - 1; i++)
        {
            if(ratings[i + 1] > ratings[i]){
                candy[i + 1] = candy[i] + 1;
            }
        }
    //从后向前,比右,左孩子大于右孩子
    for (int j = ratings.size() - 1; j > 0; j--)
        {
            if(ratings[j - 1] > ratings[j]){
                candy[j - 1] = max(candy[j - 1],candy[j] + 1);
            }
        }

    //返回统计结果
    return accumulate(candy.begin(), candy.end(), 0);
}

int main(void){
    int num;
    vector<int> ratings;
    while (cin >> num) {
        ratings.push_back(num);
        if (::getchar() == '\n') {
            break;
        }
    }

    int result;
    result = candy(ratings);
    cout << result << endl;
    return 0;
}
/*
示例1:
1 0 2
示例2:
1 2 2
 * */
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

希希雾里

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值