题干:
代码:
class Solution {
public:
int candy(vector<int>& ratings) {
vector<int> candy(ratings.size(), 1);
for(int i = 1; i < ratings.size(); i++){
if(ratings[i] > ratings[i - 1]) candy[i] = candy[i - 1] + 1;
}//比低分多一个
for(int i = ratings.size() - 2; i >= 0; i--){
if(ratings[i] > ratings[i + 1]) candy[i] =max(candy[i], candy[i + 1] + 1);
}//左边可能经过上一轮遍历变得非常大,比如5——2,而5>2+1,所以需要进行比较。
int res = 0;
for(int i = 0; i < candy.size(); i++){
res += candy[i];
}
return res;
}
};
贪心思路:分别从左到右(右边比左边大则右边比左边多一个)再从右到左(左边比右边大),rating更高比低分多1个。
易错点:假设右边比左边大,右边并不是+=1,因为一开始都设置为1,但随着遍历,假设遍历前两位后:1——2——1——1...,此时第三位比第二位大的话就得是2+1=3而不是1+1,注意题干要求是比左边多一个而不是比自己多一个。