算法第十三周作业01

Description

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

• 首先理解题意：如果该孩子的权值比傍边孩子的大，那么他的糖果数比傍边孩子的要多，对于权值为[1,2,2]情况，第一个孩子糖果数为1；第二个孩子权值比第一个孩子权值大，故为2；第三个孩子糖果树不比前面孩子（只有前面邻居）多，故糖果树为1
• 分别考虑权值的正向上升和反向上升。上升过程，如果对应糖果数arr[i]比待比较糖果数arr[i-1]大，则arr[i]保持不变，arr[i]=arr[i-1]+1。
• 对于权值序列[1,2,3,6,5,2,3,3,4]，首先输出化糖果树序列[0,0,0,0,0,0,0,0,0] （PS：题目中要求每个孩子至少1个糖果，为避免初始化序列浪费O(n)复杂度，此处不进行手动初始化，有后面统一加上序列长度来弥补），对于正向上升，糖果树序列为[0,1,2,3,0,0,1,0,1]，对于反向上升，糖果序列变换为[0,1,2,3,1,0,1,0,1]

Code

public class Solution {
    public int candy(int[] ratings) {
        int result = 0;
        int[] arr = new int[ratings.length];
        // 正向上升
        for (int i = 1; i < ratings.length; i++) {
            // 相邻对比
            if(ratings[i] > ratings[i-1]){
                arr[i] = arr[i-1] + 1;
            }
        }
        // 反向上升
        for (int i = ratings.length - 1; i > 0; i--) {
            // 相邻对比，如果权值递增时，糖果数也是递增的，则忽略操作
            if(ratings[i-1] > ratings[i] && arr[i-1] <= arr[i]){
                arr[i-1] = arr[i] + 1;
            }
            // 叠加结果
            result += arr[i];
        }
        return result + arr[0] + ratings.length;
    }
}