candy
题目描述
有N个小朋友站在一排,每个小朋友都有一个评分,你现在要按以下的规则给孩子们分糖果:
每个小朋友至少要分得一颗糖果,分数高的小朋友要他比旁边得分低的小朋友分得的糖果多
你最少要分发多少颗糖果?
题目解析
题目要求每个小孩至少一颗,相邻的小孩分数高的比分数低的分的多,并且要最少分配糖果,所以我们可以分三步分配:
1、每个小孩分配一颗
2、从左到有依次遍历,相邻的分数高的比分数低的多加一颗(这一步保证分数高的比分数低的分到的糖果多)
3、从右边到左 分数高的比分数低的多加一颗(这一步为了保证最少分配)
原题地址
代码
import java.util.Arrays;
public class Solution {
public int candy(int[] ratings) {
//首先保证每一个小孩都有一个
//正一次反一次保证相邻的分数高的比分数低的分的糖果多
int sum = 0;
int []count = new int[ratings.length];
Arrays.fill(count,1);//每一个小孩初始都一个
for(int i = 1; i < ratings.length ; i++){//正向
if(ratings[i-1] < ratings[i]){//前一个分数比当前的低
count[i] = count[i-1] + 1;
}
}
for(int j = ratings.length -1 ; j > 0 ; j--){//反向
sum = sum + count[j];
if(ratings[j] < ratings[j-1] && count[j] >= count[j-1]){
count [j-1] = count[j]+1;//为了更少的分配
}
}
sum = sum + count[0];
return sum;
}
}