lintcode-分糖果

空間復雜度O(n)

class Solution {
public:
    /**
     * @param ratings Children's ratings
     * @return the minimum candies you must give
     */
    int candy(vector<int>& ratings) {
        // Write your code here
        
        vector<int> dp(ratings.size(), 1);
        
        for(int i = 1; i < dp.size(); ++i){
            if(ratings[i] > ratings[i-1]){
                dp[i] = max(dp[i], dp[i-1] + 1);
            }
        }
        
        for(int i = dp.size() - 1; i > 0; --i){
            if(ratings[i] < ratings[i-1]){
                dp[i-1] = max(dp[i-1], dp[i] + 1);
            }
        }
        
        return accumulate(dp.begin(), dp.end(), 0);
    }
};

空間復雜度O(1)

class Solution {
public:
    /**
     * @param ratings Children's ratings
     * @return the minimum candies you must give
     */
    int candy(vector<int>& ratings) {
        // Write your code here
        
        int total = 0;
        int preCount = 1;  //前一個小孩的糖果數
        int descendStartCount = preCount;  //遞減序列開始的孩子的糖果數
        int length = 0;  
        
        if(ratings.size() == 0) {
            return 0;
        }
        
        total++;
        for(int i = 1; i < ratings.size(); ++i) {
            if(ratings[i] < ratings[i-1]) {
                length ++ ;
                if(descendStartCount <= length) {
                    total++;  //根據遞減序列的長度修正遞減序列第一個孩子的糖果數
                }
                total += length;  //逆序獲取后續孩子的糖果數
                preCount = 1;     //注意
            }else{
                int currentCount = 0;
                if(ratings[i] > ratings[i-1]) {
                    currentCount = preCount + 1;
                }else{
                    currentCount = 1;
                }
                
                total += currentCount;
                preCount = currentCount;
                length = 0;      //注意
                descendStartCount = currentCount;
            }
        }
        
        return total;
        
    }
};
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容