思路:根据题意,分发糖果应满足从左到右遍历时,第i个孩子分数高于i-1个孩子,则糖果个数加1;从右往左遍历时,第i个孩子分数高于i+1个孩子,则糖果个数加1,最后去两次遍历结果的最大值,得到结果。
class Solution {
public:
int candy(vector<int>& ratings) {
int n=ratings.size();
vector<int>left(n,1);
vector<int>right(n,1);
for(int i=1;i<n;i++){
if(ratings[i]>ratings[i-1])
left[i]=left[i-1]+1;
}
int res=left[n-1];
for(int i=n-2;i>=0;i--){
if(ratings[i]>ratings[i+1])
right[i]=right[i+1]+1;
res+=max(left[i],right[i]);
}
return res;
}
};