135. 分发糖果
描述
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
链接
135. 分发糖果 - 力扣(LeetCode) (leetcode-cn.com)
解法:贪心
采用了两次贪心的策略:
- 一次是从左到右遍历,只比较右边孩子评分比左边大的情况。
- 一次是从右到左遍历,只比较左边孩子评分比右边大的情况。
1 class Solution { 2 public int candy(int[] ratings) { 3 /** 4 分两个阶段 5 1、起点下标1 从左往右,只要 右边 比 左边 大,右边的糖果=左边 + 1 6 2、起点下标 ratings.length - 2 从右往左, 只要左边 比 右边 大,此时 左边的糖果应该 取本身的糖果数(符合比它左边大) 和 右边糖果数 + 1 二者的最大值,这样才符合 它比它左边的大,也比它右边大 7 */ 8 int[] candy = new int[ratings.length]; 9 candy[0] = 1; 10 for (int i = 1; i < ratings.length; i++) { 11 if (ratings[i] > ratings[i - 1]) { 12 candy[i] = candy[i - 1] + 1; 13 } else { 14 candy[i] = 1; 15 } 16 } 17 18 for (int i = ratings.length - 2; i >= 0; i--) { 19 if (ratings[i] > ratings[i + 1]) { 20 candy[i] = Math.max(candy[i], candy[i + 1] + 1); 21 } 22 } 23 24 int SumCandy = 0; 25 for (int s : candy) { 26 SumCandy += s; 27 } 28 return SumCandy; 29 } 30 }
参考
carl