LeetCode 135.Candy （分發糖果）

題目描述：

老師想給孩子們分發糖果，有 N 個孩子站成了一條直線，老師會根據每個孩子的表現，預先給他們評分。

你需要按照以下要求，幫助老師給這些孩子分發糖果：

• 每個孩子至少配置設定到 1 個糖果。
• 相鄰的孩子中，評分高的孩子必須獲得更多的糖果。

那麼這樣下來，老師至少需要準備多少顆糖果呢？

示例 1:

輸入: [1,0,2]
輸出: 5
解釋: 你可以分别給這三個孩子分發 2、1、2 顆糖果。
      

示例 2:

輸入: [1,2,2]
輸出: 4
解釋: 你可以分别給這三個孩子分發 1、2、1 顆糖果。
     第三個孩子隻得到 1 顆糖果，這已滿足上述兩個條件。      

AC  C++  Solution:

通過兩次周遊，一次周遊比較前一個元素，一次周遊比較後一個元素。

class Solution {
public:
    int candy(vector<int>& ratings) {
        int size = ratings.size();
        if(size <= 1)
            return size;
        
        vector<int> num(size,1);
        for(int i = 1; i < size; i++) {     //正序比較
            if(ratings[i] > ratings[i-1])
                num[i] = num[i-1] + 1;
        }
        
        for(int i = size-1; i > 0; i--) {   //逆序比較
            if(ratings[i-1] > ratings[i])
                num[i-1] = max(num[i]+1,num[i-1]);
        }
        
        int result = 0;
        for(auto i : num)
            result += i;
        
        return result;
    }
};