贪心算法 LeetCode135 Candy(hard)

题目描述

一群孩子站成一排，每一个孩子有自己的评分。现在需要给这些孩子发糖果，规则是如果一个孩子的评分比自己身旁的一个孩子要高，那么这个孩子就必须得到比身旁孩子更多的糖果；所有孩子至少要有一个糖果。求解最少需要多少个糖果。

输入输出样例
输入是一个数组，表示孩子的评分。输出是最少糖果的数量。

Input: [1,0,2]
Output: 5

在这个样例中，最少的糖果分法是 [2,1,2]。

题解1

两次遍历：
思路：两次遍历分别满足左右规则。
1.从左向右遍历时，保证序列中，右边分数较高的比相邻左边的糖果数多一。即：如果ratings[i]>ratings[i-1]，则 candy[右=[i]=candy[i-1]+1 .此步骤保证了对于任意两个元素AB，若B>A，则B糖果数>A糖果数。
2.从右向左遍历时，保证序列中，左边分数较高的比相邻右边的糖果数多。如果ratings[i-1]>ratings[i]，并且candy[i-1]<=candy[i],则candy[左]=candy[右]+1

提示：为什么从左往右遍历时，当右边比左边大时更新右边，而不是选择当左边比右边大时更新左边？

因为从左往右遍历，更新右边的值时，在下次循环时，能用到上次更新的值。

Java代码：

public int candy(int[] ratings) {
    	int numOfCandy = ratings.length;
    	int[] record = new int[ratings.length];
    	int cnt1 = 0;
    	int cnt2 = 0;
    	
    	for(int i = 0; i < ratings.length-1; i++) {
    		if(ratings[i+1] > ratings[i]) {
    			numOfCandy += record[i]-record[i+1]+1;
    			record[i+1] = record[i]+1;
    		}
    	}
    	
    	for(int i = ratings.length-1; i > 0; i--) {
    		if(ratings[i] < ratings[i-1] && record[i-1] <= record[i]) {
    			numOfCandy += record[i] - record[i-1] + 1;
    			record[i-1] = record[i] + 1;
    		}
    	}
    	
    	return numOfCandy;
 }

题解2

一次遍历：
思路：

当为升序时：后一个糖果数等于前一个+1
当为降序时：糖果增量为[降序序列长度]；当降序序列长度大于增序序列长度时，增序序列最后一个也加入降序序列长度。
即：需要一个pre记录上一个人的糖果数，dec记录降序序列长度，ans记录总糖果数。

Java:

public int candy(int[] ratings) {
    	int numOfCandy = ratings.length;
    	int[] record = new int[ratings.length];
    	int cnt1 = 0;
    	int cnt2 = 0;
    	
    	for(int i = 0; i < ratings.length-1; i++) {
    		if(ratings[i+1] > ratings[i]) {
    			numOfCandy += record[i]-record[i+1]+1;
    			record[i+1] = record[i]+1;
    		}
    	}
    	
    	for(int i = ratings.length-1; i > 0; i--) {
    		if(ratings[i] < ratings[i-1] && record[i-1] <= record[i]) {
    			numOfCandy += record[i] - record[i-1] + 1;
    			record[i-1] = record[i] + 1;
    		}
    	}
    	
    	return numOfCandy;
 }