Candy [leetcode] O(n)时间复杂度，O(1)空间复杂度的方法

最新推荐文章于 2022-02-17 02:41:32 发布

转载最新推荐文章于 2022-02-17 02:41:32 发布 · 625 阅读

文章标签：

#leetcode #candy

算法及程序专栏收录该内容

45 篇文章

订阅专栏

本文介绍了一种分配糖果的优化算法，通过两次遍历数组，确保每个糖果的数量基于其右侧和左侧的评级进行合理分配。

首先初始化，每个人分得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;
}