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

来源:互联网 发布:硬笔入门字帖 知乎 编辑:程序博客网 时间:2024/05/16 11:06

首先初始化,每个人分得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;
}
0 0
原创粉丝点击