[LeetCode] Candy

標籤：style   class   blog   code   color   get   

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?

例子：

input ： 　　　  1 1 7 6 5 9 7

第一次掃描：　　1 1 2 1 1 2 1

第二次掃描：　　1 1  3  2  1 2 1

 1 class Solution { 2 public: 3     int candy(vector<int> &ratings) { 4         int res = 0; 5         int n = ratings.size(); 6         if (n == 0) { 7             return res; 8         } 9         int *t = new int[n];10         for (int i = 0; i < n; ++i) {11             t[i] = 1;12         }13         //從左向右掃描  ，保證當前比前一個多14         for (int i = 1; i < n; ++i) {15             if (ratings[i] > ratings[i-1]) {16                 t[i] = t[i-1] + 1;17             }18         }19         //從右向左掃描， 保證當前比後一個多     20         for(int i = n-2;i >= 0;i--)21         {             22             //如果第i個的小孩的權值比i+1個小孩高,但是糖的數目卻小或者相等，那麼i的糖數目等於i+1的糖數目+1。             23             if(ratings[i] > ratings[i+1] && t[i] <= t[i+1])24             {                 25                 t[i] = t[i+1] + 1;            26             }      27         }28         for (int i = 0; i < n; ++i) {29             res += t[i];30         }31         return res;32     }33 };