题目
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
输入: [1,0,2]
输出: 5
解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
输入: [1,2,2]
输出: 4
解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
思路
注意:评分相等的话可以一个多一个少,比如两个评分都是[2,2],但是糖果可以[1,2],另外分数大加一个糖果就行,但是分数小可以直接只给1个,因为要用最少的糖果
本题要先确定一边,再确定另一边
局部最优 :只要右边评分比左边大,右边的孩子就多一个糖果
全局最优 :相邻的孩子中,评分高的右孩子获得比左边孩子更多的糖果
1、先确定右边分数比左边分数高的情况(从前向后遍历):
如果score[i] > score[i-1]
,那么score[i]
的糖果一定要比score[i-1]
多一个,即贪心为candyNum[i] = candyNum[i-1] +1
2、再确定左边分数比右边分数高的情况(从后往前遍历):这里不能从前往后遍历,因为如果从前向后遍历,根据 ratings[i + 1]
来确定 ratings[i]
对应的糖果,那么每次都不能利用上前一次的比较结果了。
如果score[i] > score[i+1]
,此时cangyNum[i]
就有两个选择了,一个是candyNum[i+1]+1
(从右边这个加一得到的糖果数量),一个是candyNum[i]
(之前比较右孩子大于左孩子得到的糖果数量)
这里进行第二次贪心:
局部最优:取candyNum[i+1]+1
和candyNum[i]
中最大的糖果数量,保证第i
个小孩的糖果数既大于左边也大于右边
全局最优:相邻的孩子中,评分高的孩子获得更多的糖果。
java代码如下:
class Solution{
public int candy(int[] scores){
int len = scores.length;
int[] candyNum = new int[len];
candyNum[0] = 1;//因为最开始是从前往后遍历,并且每个孩子最少分到一个糖果
for(int i = 1; i < len; i++){//从前往后遍历,先判断右边比左边高的情况
candyNum[i] = (scores[i] > scores[i-1]) ? candyNum[i-1] + 1 : 1;
}
for(int i = len -2; i >= 0; i--){//从后往前遍历,继续判断左边比右边高的情况
if(scores[i] > scores[i + 1]){
candyNum[i] = Math.max(candyNum[i],candyNum[i+1]+1);
}
}
int ans = 0;
for(int num : candyNum){
ans += num;
}
return ans;
}
}