https://leetcode.com/problems/candy/description/
这个题描述的很烂,但是其实很值得做。
题目:n个孩子,每个孩子一个rating值, 每个孩子至少一个糖,如果这个孩子的rating比邻居高,那么这个孩子的糖的个数必须大于邻居。
输出:最少要准备多少个糖发给所有孩子。
方法:顺序走一遍,然后逆序走一遍。
逆序的时候因为是ans[i] = max(ans[i+1] + 1, ans[i])。所以乜有破坏顺序走的那一边的性质,也就是ans[i] > ans[i-1] if ratings[i] > ratings[i-1]
class Solution {
public:
int candy(vector<int>& ratings) {
vector <int> ans(ratings.size(), 1);
int sum = 0;
for (int i = 1; i < ratings.size(); i ++) {
if (ratings[i] > ratings[i-1])
ans[i] = ans[i-1] + 1;
}
sum = ans[ratings.size()-1];
for (int i = ratings.size() -2 ; i >=0; i --) {
if (ratings[i] > ratings[i+1]) {
ans[i] = max(ans[i+1] + 1, ans[i]);
}
sum += ans[i];
}
return sum;
}
};