/**135. Candy
* @param ratings
* @return
*/
public int candy(int[] ratings) {
int len = ratings.length;
if (len == 0)
return 0;
int[] left = new int[len];
int[] right = new int[len];
left[0] = 1;
for (int i=1; i<len; i++) {
if (ratings[i] > ratings[i-1])
left[i] = left[i-1]+1;
else
left[i] = 1;
}
right[len-1] = left[len-1];
for (int i=len-2; i>=0; i--) {
if (ratings[i] > ratings[i+1])
right[i] = right[i-1]+1;
else
right[i] = 1;
}
int sum = 0;
for (int i=0; i<len; i++) {
sum += (left[i]>right[i])? left[i]: right[i];
}
return sum;
}
这道题和Trapping water那个是一样的想法,因为无论是水坑还是得到糖的小朋友,影响因素都不只一边,都是左右两边的最小值/最大值来决定的。
所以这道题跟上一道一样,也是左右两边遍历数组。
left数组存从左边遍历,当前小朋友对比其左边小朋友,他能拿到糖的数量;
right数组存从右边遍历,当前小朋友对比其右边小朋友,他能拿到的糖的数量。
最后针对这两个数组,每个小朋友能拿到的糖的数量就是这两个数最大的那个数,求总加和就好了。
稍稍改进一下,在遍历right时直接求和
public int candy(int[] ratings) {
int len = ratings.length;
if (len == 0) {
return 0;
}
int[] nums = new int[len];
nums[0] = 1;
for (int i = 1; i < len; i++) {
if(ratings[i] > ratings[i-1]) {
nums[i] = nums[i-1] + 1;
} else {
nums[i] = 1;
}
}
int res = nums[len-1];
for (int i = len-2; i >= 0; i--) {
int cur = 1;
if(ratings[i]>ratings[i+1])
{
cur = nums[i+1]+1;
}
res += Math.max(cur,nums[i]);
nums[i] = cur;
}
return res;
}