leetcode 135分发糖果(贪心算法)
题目:
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。 你需要按照以下要求,帮助老师给这些孩子分发糖果: 每个孩子至少分配到 1 个糖果。 评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。 那么这样下来,老师至少需要准备多少颗糖果呢?
思路:
最开始的错误思路:每个孩子初始糖果记为1,在得分里找到分数最小的孩子,记录下标min,
从min开始,往前遍历到0,如果当前坐标i 孩子的得分比下一个坐标i+1 孩子的得分高,当前孩子糖果数目记为candy[i+1]+1,即只多一个糖果;
从min开始,往后遍历到末尾,如果当前坐标i+1 孩子的得分比上一个坐标i 孩子的得分高,当前孩子糖果数目记为candy[i]+1,即只多一个糖果。
错误的地方:题目中指明,评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果,这种遍历都只与单侧的孩子得分进行比较。
正确思路:
不需要找到分数最少的孩子
首先从前往后遍历,保证每一个孩子都与他左侧的孩子进行比较,若分数高,则糖果数为左侧孩子糖果数+1,
然后从后往前遍历,保证每一个孩子都与他右侧的孩子进行比较,若分数高,则糖果数为右侧孩子糖果数+1,(此时注意:已经经过一轮遍历了,可能分数高的孩子,糖果数已经比右侧孩子的糖果数多了,此时,不需要改变糖果数目)
class Solution {
public:
int candy(vector<int>& ratings) {
int cnt=ratings.size();
vector<int>candys(cnt,1);
for(int i=0;i<cnt-1;i++){
if(ratings[i]<ratings[i+1]){
candys[i+1]=candys[i]+1;
}
}
for(int j=cnt-1;j>0;j--){
if(ratings[j]<ratings[j-1] && candys[j-1]<=candys[j]){
candys[j-1]=candys[j]+1;
}
}
int num=0;
for(int i=0;i<candys.size();i++){
num+=candys[i];
}
return num;
}
};
结论:贪心算法是先找到局部最优解,即最用少的资源满足最容易满足的条件,再逐步扩大问题规模进行解决