135. candy 贪心算法

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?

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

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

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

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

public class Solution {
    public int candy(int[] rating) {
        int len=rating.length;
        int [] candy=new int[len];
        for(int i=0;i<len;i++){
            candy[i]=1;
        }
        for(int i=1;i<len;i++){
            if(rating[i]>rating[i-1])
                candy[i]=candy[i-1]+1;
        }
        for(int i=len-2;i>=0;i--){
            if((rating[i]>rating[i+1])&&(candy[i]<=candy[i+1]))
                candy[i]=candy[i+1]+1;
        }
        int num=0;
        for(int i=0;i<len;i++){
            num+=candy[i];
        }
        return num;
    }
}
    原文作者:贪心算法
    原文地址: https://blog.csdn.net/u012985132/article/details/52529199
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞