分發糖果(Java實現)

題目

老師想給孩子們分發糖果,有 N 個孩子站成了一條直線,老師會根據每一個孩子的表現,預先給他們評分。
你須要按照如下要求,幫助老師給這些孩子分發糖果:
每一個孩子至少分配到 1 個糖果。
相鄰的孩子中,評分高的孩子必須得到更多的糖果。
那麼這樣下來,老師至少須要準備多少顆糖果呢?數組

來源:力扣(LeetCode)
連接:https://leetcode-cn.com/problems/candycode

示例1

輸入: [1,0,2]
輸出: 5
解釋: 你能夠分別給這三個孩子分發 二、一、2 顆糖果。leetcode

示例2

輸入: [1,2,2]
輸出: 4
解釋: 你能夠分別給這三個孩子分發 一、二、1 顆糖果。
第三個孩子只獲得 1 顆糖果,這已知足上述兩個條件get

解題思路

兩個數組left和right分別記錄從左向右規則和從右向左規則時,每一個孩子應該分的糖果數。
題目要求即爲同時知足兩個方向的規則,即兩個數組按位取max。
最終結果要加上每一個孩子至少一個糖果。(或者left,right數組初始化爲1)io

class Solution {
    public int candy(int[] ratings) {
        int len = ratings.length;
    	int sum = len;
    	int[] left = new int[len];
    	int[] right = new int[len];
    	for (int i = 1; i < len; i++) {
            if(ratings[i] > ratings[i-1]) {
                left[i] = left[i-1]+1;
            }
        }
    	for (int i = len-2; i>=0; i--) {
            if(ratings[i] > ratings[i+1]) {
                right[i] = right[i+1]+1;	
            }
        }
    	
    	for (int i = 0; i < len; i++) {
            sum += Math.max(left[i], right[i]);
        }
    	return sum;
    }
}
相關文章
相關標籤/搜索