https://leetcode.cn/problems/candy/
题目要求
n
个孩子站成一排。给你一个整数数组 ratings
表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
- 每个孩子至少分配到
1
个糖果。 - 相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
贪心:两次遍历
class Solution {
public int candy(int[] ratings) {
int n = ratings.length;
int[] candy = new int[n];
candy[0] = 1;
// Arrays.fill(candy, 1);
for (int i = 1; i < n; i++) {
if (ratings[i] > ratings[i - 1])
candy[i] = candy[i - 1] + 1;
else
candy[i] = 1;
}
for (int i = n - 2; i >= 0; i--) {
if (ratings[i] > ratings[i + 1])
candy[i] = Math.max(candy[i], candy[i + 1] + 1);
}
int res = 0;
for (int i : candy) {
res += i;
}
return res;
}
}
- 贪心的策略,如果在考虑局部的时候想两边兼顾,就会顾此失彼。所以采用两次贪心的策略:
- 一次是从左到右遍历,只比较右边孩子评分比左边大的情况。
- 一次是从右到左遍历,只比较左边孩子评分比右边大的情况。
这样从局部最优推出了全局最优,即:相邻的孩子中,评分高的孩子获得更多的糖果。