思路:
两个条件
每个孩子至少分配到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;
- }
- }