Candy

There are N children standing in a line. Each child is assigned a rating value.

You are giving candies to these children subjected to the following requirements:

  • Each child must have at least one candy.
  • Children with a higher rating get more candies than their neighbors.

What is the minimum candies you must give?

Solution

从左往右遍历一次,如果比左边的值大,就在左边的值的基础上加一。然后再从右往左遍历一次,如果比右边的值大,就在右边的值的基础上加一

Complexity

  1. O(n) space.
  2. traverse only once with O(1) space.

Code

public class Solution {
    public int candy(int[] ratings) {
        return candy_1(ratings);
    }
    
    public int candy_1(int[] ratings) {
        int N = ratings.length;
        if (N == 0) return 0;
        int[] height = new int[N];
        int res = 0;
        height[0] = 1;
        for (int i = 1; i < N; ++i) {
            height[i] = 1;
            if (ratings[i] > ratings[i - 1]) {
                height[i] = height[i - 1] + 1;
            }
        }
        for (int i = N - 2; i >= 0; --i) {
            if (ratings[i] > ratings[i + 1]) {
                height[i] = Math.max(height[i], height[i + 1] + 1);
            }
        }
        for (int i = 0; i < N; ++i) {
            res +=height[i];
        }
        return res;
    }
    
    public int candy_2(int[] ratings) {
        int N = ratings.length;
        if (N == 0) return 0;
        int candy = 1, res = 1;
        int maxVal = 1, maxIdx = 0;
        for (int i = 1; i < N; ++i) {
            if (ratings[i] >= ratings[i - 1]) {
                candy = ratings[i] == ratings[i - 1] ? 1 : candy + 1;
                maxVal = candy;
                maxIdx = i;
            } else {
                if (candy == 1) {
                    if (maxVal <= i - maxIdx) {
                        ++maxVal;
                        ++res;
                    }
                    res += i - maxIdx - 1;
                }
                candy = 1;
            }
            res += candy;
        }
        return res;
    }
}