有 N 个小孩站成一列。每一个小孩有一个评级。
依照下面要求。给小孩分糖果:
-
每一个小孩至少得到一颗糖果。
-
评级越高的小孩能够得到很多其它的糖果。
需最少准备多少糖果?
您在真实的面试中是否遇到过这个题?Yes
例子
标签 Expand 给定评级 = [1, 2]
, 返回 3
.
给定评级 = [1, 1, 1]
, 返回 3
.
给定评级 = [1, 2, 2]
, 返回 4
.
([1,2,1]).
分析:事实上题目描写叙述的有点问题,不是评级越高的小孩能够得到很多其它的糖果。从第三个例子中能够看出来,应该是“相邻两人里评级高的。能够得到很多其它的糖果”,一開始能够如果每一个小孩都1颗糖果,如果第i个小孩比i-1个小孩rating高,那么糖果数也得比他高。和i+1个小孩比也是这样,于是能够从前往后扫描一遍,再从后往前扫描一遍。
代码:
class Solution { public: /** * @param ratings Children's ratings * @return the minimum candies you must give */ int candy(vector<int>& ratings) { // Write your code here vector<int> dp(ratings.size(),1); for(int i=0;i<ratings.size()-1;i++) { if(ratings[i]<ratings[i+1]) dp[i+1] = max(dp[i+1],dp[i]+1); } for(int i=ratings.size()-1;i>=1;i--) { if(ratings[i]<ratings[i-1]) dp[i-1] = max(dp[i-1],dp[i]+1); } return accumulate(dp.begin(),dp.end(),0); } };