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

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 <= 10(4)
• -100 <= nums[i] <= 100
• 1 <= k <= 10(4)

代码

package __greedy

import "sort"

func largestSumAfterKNegations(nums []int, k int) int {
   /*
      k次取反，m个负数
      k>m，m个负数全部取反，绝对值最小的数取反：(k-m)%2==0不变，!=0取反
      k==m,m个负数全部取反
      k<m,绝对值最大的数取反
   */
   n := len(nums)
   sort.Ints(nums)
   m := 0
   for m < n { //计数，负数
      if nums[m] < 0 {
         m++
      } else {
         break
      }
   }
   if k > m {
      if (k-m)%2 != 0 {//绝对值最小的数取反
         if m == n || (m-1 > 0 && nums[m] > (-nums[m-1])) {
            nums[m-1] = -nums[m-1]
         } else {
            nums[m] = -nums[m]
         }

      }
   }
   sum := 0
   for i := 0; i < n; i++ { //取反
      if i < min(m, k) {
         nums[i] = -nums[i]
      }
      sum += nums[i]
   }

   return sum
}
func min(i, j int) int {
   if i > j {
      return j
   }
   return i
}

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 升汽油
示例 2:
输入: gas = [2,3,4], cost = [3,4,3]输出: -1解释:你不能从 0 号或 1 号加油站出发，因为没有足够的汽油可以让你行驶到下一个加油站。

提示:

• gas.length == n
• cost.length == n
• 1 <= n <= 10(5)
• 0 <= gas[i], cost[i] <= 10(4)
  思路
• 那么局部最优：当前累加rest[i]的和curSum一旦小于0，起始位置至少要是i+1，因为从i之前开始一定不行
  [图片]

代码

package __greedy

func canCompleteCircuit(gas []int, cost []int) int {
   n := len(gas)
   store := make([]int, n)
   for i := 0; i < n; i++ {
      store[i] = gas[i] - cost[i]
   }
   /*
      不能以store为负的加油站为起点
   */
   for i := 0; i < n; i++ {
      if store[i] < 0 {
         continue
      }
      sum := 0
      for j := i; j < (n + i); j++ {
         sum += store[j%n]
         if sum < 0 { //一旦小于0，起始位置至少要是i+1
            i = j
            break
         }
      }
      if sum >= 0 {
         return i
      }

   }
   return -1
}

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 * 10(4)
• 0 <= ratings[i] <= 2 * 10(4)

代码

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;
    }
};