《录鼎记》——贪心算法part03

最新推荐文章于 2024-10-05 23:32:32 发布

宇文辰兴

最新推荐文章于 2024-10-05 23:32:32 发布

阅读量78

点赞数

文章标签：贪心算法算法

本文链接：https://blog.csdn.net/qq_56013387/article/details/130231081

版权

文章介绍了如何使用贪心算法解决LeetCode中的三个问题：1)通过K次取反最大化数组和，策略是按绝对值排序后从最小元素开始翻转；2)在加油站问题中，通过迭代找到可以从哪个站开始完成旅程；3)分发糖果问题，保证评分高的孩子得到更多糖果，两次遍历确保满足条件。

摘要由CSDN通过智能技术生成

今日任务：

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

134. 加油站

135. 分发糖果

一、k次取反后最大化的数组和

力扣题目链接 (opens new window)

给定一个整数数组 A，我们只能用以下方法修改该数组：我们选择某个索引 i 并将 A[i] 替换为 -A[i]，然后总共重复这个过程 K 次。（我们可以多次选择同一个索引 i。）

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

class Solution {
public:
    static bool cmp(int a ,int b){
        return abs(a) > abs(b);
    }
    int largestSumAfterKNegations(vector<int>& A, int K) {
        sort(A.begin(),A.end(),cmp);
        for(int i=0;i<A.size();i++){
            if(A[i]<0 &&K>0){
                A[i] *=-1;
                K--;
            }
        }
        if(K%2==1) A[A.size()-1] *=-1;
        int result =0;
        for(int a: A) result+=a;
        return result;

    }
};

这题的思路就是先按照绝对值大小排序，为什么要排序呢，因为我们要贪心，所以就是说要从最小的元素开始翻转。

确实比较简单，但还是需要用贪心的算法思考。

二、加油站

力扣题目链接 (opens new window)

先用了暴力算法，是超时的。

然后用贪心算法来

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int cursum =0;
        int min = INT_MAX;
        for(int i=0;i<gas.size();i++){
            int rest =gas[i]-cost[i];
            cursum+=rest;
            if(cursum<min){
                min=cursum;
            }
        }
        if(cursum<0) return -1;
        if(min>=0) return 0;
        for(int i=gas.size()-1;i>=0;i--){
            int rest =gas[i]-cost[i];
            min +=rest;
            if(min>=0){
                return i;
            }
        }
        return -1;

    }
};

方法二：

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int cursum =0;
        int totalsum =0;
        int start =0;
        for(int i=0;i<gas.size();i++){
            cursum+=gas[i]-cost[i];
            totalsum+=gas[i]-cost[i];
            if(cursum<0){
                start=i+1;
                cursum=0;
            }
        }
        if(totalsum<0) return -1;
        return start;

    }
};

局部最优是可以走完这一程，那么就是说如果小于0的时候，应该从下一个节点开始，那么从下一个节点开始呢，如果能够有足够的油量剩余，也就是totalsum的油量如果大于0，那么就是能从一个可以的节点开始往下走。

三、分发糖果

力扣题目链接 (opens new window)

老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。

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

每个孩子至少分配到 1 个糖果。
相邻的孩子中，评分高的孩子必须获得更多的糖果。

那么这样下来，老师至少需要准备多少颗糖果呢？

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 result =0;
        for(int i=0;i<candyvec.size();i++){
            result+=candyvec[i];
        }
        return result;

    }
};

总结：从左到右是处理右边孩子比左边大的情况，另一次是右往左遍历，只是比较左边孩子评分比右边大的情况。