首页 > 代码库 > 【leetcode】Candy

【leetcode】Candy

There are N children standing in a line. Each child is assigned a rating value.

You are giving candies to these children subjected to the following requirements:

  • Each child must have at least one candy.
  • Children with a higher rating get more candies than their neighbors.

What is the minimum candies you must give?


 

题解:

  1. 第一个孩子给一颗糖,然后从左到右遍历rate数组,如果孩子i+1的rate比孩子i的rate高,那么孩子i+1得到的糖果数目比孩子i得到的糖果数目多1;如果孩子i+1的rate比孩子i的rate低,那么就给孩子i+1一颗糖;
  2. 从右往左遍历rate数组,如果当前遍历的孩子i的rate比他右边的孩子的rate高,那么他得到的糖果就比他右边的孩子得到的糖果多1.
  3. 累加rate中所有的值,得到总的最少糖果数目。

代码如下:

 1 public class Solution { 2     public int candy(int[] ratings) { 3         if(ratings.length == 0) 4             return 0; 5          6         int[] count = new int[ratings.length]; 7         Arrays.fill(count, 1); 8          9         for(int i = 1;i <= ratings.length-1;i++){10             if(ratings[i]> ratings[i-1] )11                 count[i] = count[i-1] + 1; 12         }13         14         int sum = 0;15         for(int i = ratings.length-1;i >= 1;i--){16             sum += count[i];17             if(ratings[i-1] > ratings[i] && count[i-1] <= count[i])18                 count[i-1] = count[i]+ 1; 19         }20         21         return count[0] + sum;22     }23 }

在17行的循环,需要判断i-1个孩子当前获得的糖果数目是否真的比第i个孩子的少,如果真的少,才需要+1.例如:

ratings = {4,2,3,4,1}的时候,第一遍遍历得到的count数组是{1,1,2,3,1},此时从后往前遍历的时候ratings[3] > ratings[4],但是count[3]已经大于count[4]了,所以不需要更新count[3] = count[4]+1。