首先初始化,每个人分得1个糖果
然后从左向右扫描,使得比左边rating大的,分得的糖果多
从右向左扫描,使得比右边rating大的,分得的糖果多
int candy(vector<int> &ratings){
int size=ratings.size();
vector<int> num(size,1);
for(int i=1;i<size;i++)
{
if(ratings[i]>ratings[i-1])
num[i]=num[i-1]+1;
}
for(int i=size-2;i>=0;i--)
{
if(ratings[i]>ratings[i+1] && num[i]<=num[i+1])
num[i]=num[i+1]+1;
}
int sum=0;
for(int i=0;i<size;i++)
sum=sum+num[i];
return sum;
}