视频讲解:https://www.bilibili.com/video/BV1ev4y1r7wN/?vd_source=a935eaede74a204ec74fd041b917810c
这道题不是那么好想,因为评分要和两边的比,这就要规划怎么比了,如果每次都跟两边比,到最后会发现哪边都满足不了条件,这里采用的方法是从前往后遍历一次,从后往前遍历一次,判断前后大小情况
- 从前往后遍历,如果右大于左,糖果++
- 从后往前遍历,如果左大于右,糖果++
- 两次遍历取最大值,这样既可以满足从左往右的合法性,又能满足从右往左的合法性
cpp
class Solution {
public:
int candy(vector<int>& ratings) {
//初始化糖果数组
vector<int> candy(ratings.size(), 1);
//从左往右遍历
for(int i = 1; i < ratings.size(); ++i)
{
//如果右比左大,candy+1
if(ratings[i - 1] < ratings[i])
{
candy[i] = candy[i - 1] + 1;
}
}
//从右往左遍历
for(int i = ratings.size() - 2; i >= 0; --i)
{
//如果左比右大,candy+1
if(ratings[i] > ratings[i + 1])
{
candy[i] = max(candy[i + 1] + 1, candy[i]);
}
}
//计算结果
int result = 0;
for(int i = 0; i < candy.size(); ++i)
{
result += candy[i];
}
return result;
}
};