135. candy 贪心算法

2023年11月30日 235次阅读 来源: 贪心算法

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
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞