老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
-
每个孩子至少分配到 1 个糖果。
-
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
输入: [1,0,2]
输出: 5
解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
输入: [1,2,2]
输出: 4
解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
分析
题目要求糖果的最少消耗值,怎么才能在满足上面要求的情况下最少的发放糖果呢?
我们可以秉持一个策略:
- 从评分最低的孩子开始发
- 尽可能地发满足条件的最小数量
按照这个策略,我们从给评分最低的孩子一个糖果,然后按照评分依次增加的顺序发放糖果,如果这个孩子周围有人评分比他低,则让他比那个低分的孩子多一个糖果,如果没有人比他低,则发放一个糖果(老资本家了)。按照这个方式便能得到最小数量。
首先我们需要在不破坏原有队形的情况下得到评分排名,这里我们可以引入一个聚合类stu(其中存有评分和在原有队形中的序号),然后对这个聚合类按评分升序的条件排序(sort+lambda表达式)。排序成功后再按上述的方法不断分配糖果,从而得到正确答案。
//先按分数排序,再从分数最低的学生开始分糖果
struct stu{
int score;//分数
int index;//序号
};
/*
执行结果:通过
执行用时:88 ms, 在所有 C++ 提交中击败了5.22%的用户
内存消耗:19.4 MB, 在所有 C++ 提交中击败了5.02%的用户
*/
int candy(vector<int>& ratings) {
if (ratings.size() == 1) {
return 1;
}
vector<stu> vs;
for (int i = 0; i < ratings.size(); ++i) {
vs.push_back({
ratings[i], i });//放入保存有序号的vector中
}
//for (auto i : vs) {
// cout << i.score << "\t" << i.index << endl;
//}
auto sortFun = [](const stu& s1, stu& s2