135. Candy

题目大意：几个小孩站一排，每个小孩有个等级值，现在给小孩发糖，发的时候要遵守2个规则：（1）每个小孩至少一颗糖（2）两个相邻的小孩中，等级大的小孩一定比等级小的小孩糖多，求发糖的数目的最小值。对于等级序列[1, 2, 2]来说，发糖数目最小值为4，发糖序列为[1, 2, 1]。

分析：初始化所有小孩糖数目为1，从前往后扫描，如果第i个小孩等级比第i-1个高，那么i的糖数目等于i-1的糖数目+1；从后往前扫描，如果第i个的小孩的等级比i+1个小孩高,但是糖的数目却小或者相等，那么i的糖数目等于i+1的糖数目+1。该算法时间复杂度为O(N)。之所以两次扫描，即可达成要求，是因为：第一遍，保证了每一点比他左边candy更多(如果得分更高的话)。第二遍，保证每一点比他右边candy更多(如果得分更高的话)，同时也会保证比他左边的candy更多，因为当前位置的candy只增不减。

class Solution {
public:
    int candy(vector<int> &ratings)
    {
        // Note: The Solution object is instantiated only once and is reused by each test case.
        int *candyNum = new int[ratings.size()];//每个小孩的糖数目
        for(int i = 0; i < ratings.size(); i++)
            candyNum[i] = 1;
        for(int i = 1; i < ratings.size(); i++)
            if(ratings[i] > ratings[i-1])
                candyNum[i] = candyNum[i-1] + 1;
        for(int i = ratings.size()-2; i>=0; i--)
            if(ratings[i] > ratings[i+1] && candyNum[i] <= candyNum[i+1])
                candyNum[i] = candyNum[i+1] + 1;
        int result = 0;
        for(int i = 0; i < ratings.size(); i++)
            result += candyNum[i];
        delete []candyNum;
        return result;
    }
};

9.17更新

这道题目的一个陷阱是：只是要求相邻的更高的小孩比低的小孩拿糖多，但是没说两个相等的小孩该如何拿糖。

事实上当序列为1, 2, 2时，拿糖为1, 2, 1。因为题目没说 不允许相等的小孩中a小孩拿的糖不能比b小孩少。

这样的题目，就用经典的 双向扫描就好了。代码如下：

class Solution {
public:
    int candy(vector<int>& ratings) {
        int n = (int)ratings.size();
        int arr[n];
        memset(arr, 0, sizeof(arr));
        arr[0] = 1;
        for (int i=1; i<n; i++) {
            if (ratings[i]>ratings[i-1]) {
                arr[i] = arr[i-1]+1;
            } else {
                arr[i] = 1;
            }
        }
        int ret = arr[n-1];
        for (int i=n-2; i>=0; i--) {
            if (ratings[i]>ratings[i+1]&&arr[i]<=arr[i+1]) {
                arr[i] = arr[i+1]+1;
            }
            ret += arr[i];
        }
        return ret;
    }
};