思路:(1)先进行分析,找到符合一边的情况,再找符合另一边的情况。
(2)我们先确定右边评分大于左边的情况,即从前向后遍历。
此时局部最优,只要右边孩子的评分大于左边,右边孩子比左边孩子多一个糖果。全局最优,评分高的右孩子得到的比左孩子更多的糖果。(局部最优可以推出全局最优)。如果ratings[i] > ratings[i - 1],则candy[i] = candy[i-1] + 1.
(3)确定左孩子评分大于右边孩子,,要从后往前遍历,这样确保后面的糖果值可以利用到前面的糖果值。如果ratings[i] > ratings[i + 1],则这里我们贪心了,局部最优,取candy[i] 和 candy[i+1] + 1的最大值。保证第i个小孩的糖果数即大于左边也大于右边。
class Solution {
public int candy(int[] ratings) {
int[] candy = new int[ratings.length];
for (int i = 0; i < candy.length; i++) {
candy[i] = 1;
}
for (int i = 1; i < ratings.length; i++) {
if (ratings[i] > ratings[i - 1]) {
candy[i] = candy[i - 1] + 1;
}
}
for (int i = ratings.length - 2; i >= 0; i--) {
if (ratings[i] > ratings[i + 1]) {
candy[i] = Math.max(candy[i],candy[i + 1] + 1);
}
}
int count = 0;
for (int i = 0; i < candy.length; i++) {
count += candy[i];
}
return count;
}
}