Leetcode#135 Candy

原题地址

遍历所有小孩的分数

1. 若小孩的分数递增，分给小孩的糖果依次+1
2. 若小孩的分数递减，分给小孩的糖果依次-1
3. 若小孩的分数相等，分给小孩的糖果设为1

当递减序列结束时，如果少分了糖果，就补上，如果多分了糖果，就减掉

究竟补多少或减多少，这很容易计算，不啰嗦了。

时间复杂度：O(n)

代码：

int candy(vector<int> &ratings) {
  int n = ratings.size();
  int sum = 1;
  int len = 0;
  int prev = 1;

  for (int i = 1; i < n; i++) {
    if (ratings[i] == ratings[i - 1]) {
      prev = 1;
      len = 0;
      sum += prev;
    }
    else if (ratings[i] > ratings[i - 1]) {
      prev++;
      len = 0;
      sum += prev;
    }
    else if (ratings[i] < ratings[i - 1]) {
      prev--;
      len++;
      sum += prev;
      if (i == n - 1 || ratings[i] <= ratings[i + 1]) {
        sum += prev < 1 ? (1 - prev) * (len + 1) : (1 - prev) * len;
        prev = 1;
      }
    }
  }

  return sum;
}