思路:
两个条件
每个孩子至少分配到1个糖果。相邻的孩子中,评分高的孩子必须获得更多的糖果。
先比较一边,再比较另一边,两个分开考虑。
问的是"**至少"**需要准备多少。
局部:右边评分>左边评分,右边多得一个。
全局:相邻孩子中,评分高的右孩子获得比左孩子更多的糖果。
如果ratings[i] > ratings[i - 1]那么[i]的糖果一定要比[i - 1]多一个:candyVec[i] = candyVec[i - 1] + 1
左孩子大于右孩子的情况要从后往前遍历,这样我们后面要比较两次中的结果,取最大的值,从而保证这个值既大于左边也大于右边。
先从左往右,再从右往左。
//前->后
for(int i = 1; i < ratings.size();i++){
if(ratings[i] > ratings[i - 1])candyVec[i] = candyVec[i - 1]+1;
}
//后->前
for(int i = ratings.size() - 2; i >= 0; i--){
if(ratings[i] > ratings[i + 1]){
candyVec[i] = max(candyVec[i], candyVec[i + 1] + 1);
}
}
//完整:
class Solutin{
public:
int candy(vector<int>&ratings){
vector<int>candyVec(ratings.size(),1);
//第一次贪心,从左往右,只比较右孩子评分比左孩子大的情况
for(int i=1;i < ratings.size();i++){
if(ratings[i] > ratings[i - 1])candyVec[i] = candyVec[i - 1]+1;
}
//第二次贪心,从右往左,只比较左孩子评分比右边大的情况
for(int i = ratings.size() - 2; i >= 0; i--){
if(ratings[i] > ratings[i + 1]{
candyVec[i] = max(candyVec[i],candyVec[i + 1] + 1);
})
}
//统计结果
int result = 0;
for(int i = 0; i < candyVec.size();i++)result += candyVec[i];
return result;
}
}