LeetCode-with-JavaScript/src/candy/res.js at master · agb/LeetCode-with-JavaScript · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
/**
 * @param {number[]} ratings
 * @return {number}
 */
var candy = function(ratings) {
  const candies = ratings.map(() => 1);
  const len = ratings.length;
  if (len <= 1) return len;

  for (let index = 1; index < len; index++) {
    if (ratings[index] > ratings[index-1]) {
      candies[index] = candies[index-1] + 1;
    }
  }

  let sum = candies[len-1];
  for (let index = len-2; index >= 0; index--) {
    if (ratings[index] > ratings[index+1] && candies[index] <= candies[index+1]) {
      candies[index] = candies[index+1] + 1;
    }
    sum += candies[index];
  }

  return sum;
};