[LeetCode]贪心算法之糖果问题135 （java,C++）

题目

一群孩子站成一排，每一个孩子有自己的评分。现在需要给这些孩子发糖果，规则是如果一个孩子的评分比自己身旁的一个孩子要高，那么这个孩子就必须得到比身旁孩子更多的糖果；所有孩子至少要有一个糖果。求解最少需要多少个糖果。

输入输出样例

输入是一个数组，表示孩子的评分。输出是最少糖果的数量。

Input:[1,1,0]
Output:4

题解（Java）

import java.util.Arrays;
public class candy{
    public static void main(String []args){
        int ratings[] = new int[]{1,2,3,1};
        candy test = new candy();
        int a = test.findcandy(ratings);
        System.out.println(a);

    }
    public int findcandy(int ratings[]){
        int size = ratings.length;
        if(size<2){
            return size;
        }
        int []num = new int[size];
        Arrays.fill(num,1);  //保证每个孩子都有一个糖果
        //从左往右遍历，若当前的分数比左边高则该孩子多一块糖果
        for(int i = 1;i < size;i++){
            if(ratings[i]>ratings[i-1]){
                num[i] = num[i-1] +1; 
            }
        }
        for(int i = 0;i<size;i++){
            
            System.out.println(num[i]);
        }
        for (int i = size-1;i>0;i--){
            if(ratings[i]<ratings[i-1]){
                num[i-1] = Math.max(num[i-1],num[i]+1);

            }
        }
        int sum =0;
        for(int i = 0;i<size;i++){
            sum+=num[i];
            System.out.println(num[i]);
        }
        return sum;

    }

}

题解（C++）

int candy(vector<int>& ratings){
	int size = ratings.size();
    if(size<2){
        return ratings;
    }
    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); //保证当前糖果最大，防止被更新掉
        }
    }
    return accumulate(num.begin(),num.end(),0);
}
}