class Solution {
public int candy(int[] ratings) {
//两次局部最优 左向右贪心一次 右向左贪心一次
int[] candyVec = new int[ratings.length];
// 计算第一次贪心 记录在candyVec 比左边大就行(即+1)
candyVec[0] = 1;
for (int i = 1; i < ratings.length; i++) {
if (ratings[i] > ratings[i - 1]) {
candyVec[i] = candyVec[i - 1] + 1;
} else {
candyVec[i] = 1;
}
}
//第二次贪心 比右边大就行 需要 从尾部开始 因为这样才能利用前面计算好的最终值 (即i 利用 i+1确定的最终值)
// 为什么i 从 ratings.length - 2 因为ratings[i] > ratings[i + 1]
for (int i = ratings.length - 2; i >= 0; i--) {
if (ratings[i] > ratings[i + 1]) {
candyVec[i] = Math.max(candyVec[i], candyVec[i + 1] + 1);
}
}
int sumCandy = 0;
for (int i = 0; i < candyVec.length; i++) {
sumCandy += candyVec[i];
}
return sumCandy;
}
}
难点:两次贪心,第一次从前向后贪,记录出candyVec数组。第二次贪心从后向前,根据已有的candyVec数组重新计算candyVec数组。