代码随想录算法训练营第day33|1005.K次取反后最大化的数组和 、134. 加油站、135. 分发糖果

a.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] 。

思路:根据贪心策略,尽量选择“最小”的数取反。优先选择绝对值最大的数取反,如果此时负数取反完毕,而取反次数k仍然>0:
情况1.若k为偶数,按照规则可以选择同一个数多次反,该数仍是正数。
情况2.若k为奇数数,选择小的正数取反

代码细节:先将数组nums[i]按照绝对值从大到小排序,遍历数组;

如果nums[i]<0且k>0,将其取反;

同时求和;

最后处理一下前面的如果此时负数取反完毕,而取反次数k仍然>0时,由于前面循环里加了一次末尾的最小正数,这里实际应该减去它的2倍

class Solution {
public:
    static bool cmp(int a, int b){
        return abs(a)>abs(b);
    }
    int sum=0;//记录总和
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end(),cmp);//按绝对值从大到小排列
        for(int i=0;i<nums.size();i++){
            if((nums[i]<0)&&k>0){
                nums[i]=-nums[i];
                k--;
            }
            sum+=nums[i];//整个数组求和
        }
        if(k%2==1) sum-=2*nums[nums.size()-1];//减去两倍,因为上边当正数加了,实际总和应该是sum=sum0-a,上边是sum0+a,所以这里-2a
        return sum;
    }
};

b.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 可为起始索引。

思路:首先如果总油量减去总消耗大于等于零那么一定可以跑完一圈,否则无论从哪里出发均不能跑完;

其次,剩油量(gas[i] -cost[i])从0开始累加,如果遇到当前总量rest[i]<0,起点肯定只能在i+1及之后(不然走不完)

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int curSum=0;//当前剩余油量
        int totalSum=0;//总剩油量
        int startPos=0;//起点
        for(int i=0;i<gas.size();i++){
            curSum+=gas[i]-cost[i];
            totalSum+=gas[i]-cost[i];
            if(curSum<0){
                //说明如果起点在这及之前肯定不够油开全程
                curSum=0;//剩余油量清零,重新开始算
                startPos=i+1;//更新起点位置:只能在i之后
            }
        }
        if(totalSum<0) return -1;//如果总剩余油量<0,肯定不能跑完一圈
        return startPos;
    }
};

c.135. 分发糖果

题目连接

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

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

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

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

示例 1:

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

示例 2:

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

思路:分别从左往右和从右往左遍历,只要右边评分比左边大,右边的孩子就多一个糖果,反之亦然。

class Solution {
public:
    int candy(vector<int>& ratings) {
        vector<int> candySum(ratings.size(),1);
        //从左边开始比较相邻
        for(int i =1;i<ratings.size();i++){
            if(ratings[i]>ratings[i-1]){
                candySum[i]=candySum[i-1]+1;
                // cout<<"i1="<<i<<endl;
            }
        }
        //从右边开始比较相邻
        for(int i=ratings.size()-2;i>=0;i--){
            if(ratings[i]>ratings[i+1]){
                candySum[i]=max(candySum[i+1]+1,candySum[i]); 
                // cout<<"i2="<<i<<endl;
            }
        }
        int sum=0;//总糖果求和
        for(int i =0;i<candySum.size();i++){
            sum+=candySum[i];
        }
        return sum;
    }
};

  • 18
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值