分发糖果

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第14天，点击查看活动详情

135. 分发糖果 - 力扣（LeetCode）

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

思路

本题可以通过可以通过查找数组峰值来解答。我们可以把数组划分成数段单调增或单调减的数组（不包含峰值），划分后的数字长度用l表示，这个数字所需糖果数就是 2 +3 + ... + l。我们还需要处理一下最值，如果是最大值，取值应该是相邻两个数组长度的最大值+1，如果是最小值，取值1。
具体到解题上，定义sum表示需要的糖果数，ascCount 和 descCount分别记录升序和降序长度，prev 记录上一个同学拿到的糖果数量，比较数组相邻两个值大小，

• 如果 ratings[i]>ratings[i-1]，prev++， descCount = 0，ascCount++，sum += prev
• 如果 ratings[i]==ratings[i-1]，ascCount = 1，descCount = 0，prev = 1，sum += prev
• 如果 ratings[i]< ratings[i-1]，descCount++，prev = 1，sum += descCount

解题

/**
 * @param {number[]} ratings
 * @return {number}
 */
var candy = function (ratings) {
  const n = ratings.length;
  let sum = 1;
  let desc = 0;
  let asc = 1;
  let prev = 1;
  for (let i = 1; i < n; i++) {
    if (ratings[i] >= ratings[i - 1]) {
      desc = 0;
      prev = ratings[i] === ratings[i - 1] ? 1 : prev + 1;
      sum += prev;
      asc = prev;
    } else {
      desc++;
      if (desc === asc) {
        desc++;
      }
      sum += desc;
      prev = 1;
    }
  }
  return sum;
};