135. Candy
There are N children standing in a line. Each child is assigned a rating value.
You are giving candies to these children subjected to the following requirements:
- Each child must have at least one candy.
- Children with a higher rating get more candies than their neighbors.
What is the minimum candies you must give?
解:
这个题难度是hard,其实比较简单,开一个candy[],从左到右,从右到左遍历一次就搞定了,第一次遍历时,如果后一个的权值大于前一个,则candy数也要大一个,反之,candy数等于1。
第一次遍历后,candy[]中还存在一种违反规则的情况:ratings[i] < ratings[i-1] && candy[i-1] <= candy[i],反向遍历时依次修改。
代码:
class Solution {
public:
int candy(vector<int>& ratings) {
int size = ratings.size();
int *candy = new int[size];
candy[0] = 1;
for (int i = 1; i < size; i++) {
if (ratings[i] > ratings[i-1]) candy[i] = candy[i-1] + 1;
else candy[i] = 1;
}
for (int i = size-1; i > 0; i--) {
if (ratings[i] < ratings[i-1] && candy[i-1] <= candy[i])
candy[i-1] = candy[i] + 1;
}
int count = 0;
for (int i = 0; i < size; i++) count += candy[i];
return count;
}
};