c_cpp 糖果http://oj.leetcode.com/problems/candy/

Posted 2021-05-14

class Solution {
public:
    int candy(vector<int> &ratings) {
        int result = 0;
        vector<int> candies(ratings.size(), 1);
        for(int i = 1; i < ratings.size(); i++){
            if(ratings[i] > ratings[i - 1]){
                candies[i] = candies[i - 1] + 1;
            }
        }
        for(int i = ratings.size() - 2; i >= 0; i--){
            if(ratings[i] > ratings[i + 1]){
                int candy_temp = candies[i + 1] + 1;
                if(candy_temp > candies[i]){
                    candies[i] = candy_temp;
                }
            }
        }
        for(int i = 0; i < candies.size(); i++){
            result += candies[i];
        }
        return result;
    }
};