题目:
题目大意:几个小孩站一排,每个小孩有个等级值,现在给小孩发糖,发的时候要遵守2个规则:
(1)每个小孩至少一颗糖
(2)两个相邻的小孩中,等级大的小孩一定比等级小的小孩糖多,求发糖的数目的最小值。(等级相同的可以有多有少)
解法1:
刚刚开始所有小孩的糖果数都为1,从左往右,arr[i]>arr[i-1]则当前i的小孩糖果数加1,再从右往左,arr[i]<arr[i+1]则当前i的小孩糖果数加1,时间复杂度O(n),空间复杂度O(n)。
代码:
public static int candy1(int[] arr){
int len = arr.length;
int[] res = new int[len];
for(int i = 0; i < len; i++){
res[i] = 1;
}
for(int i = 1; i < len; i++){
if(arr[i] > arr[i-1]){
res[i] = res[i-1]+1;
}
}
for(int i = len-2; i >= 0; i--){
if(arr[i] > arr[i+1]){
res[i] = res[i+1]+1;
}
}
int sum = 0;
for(int i = 0; i < len; i++){
sum += res[i];
}
return sum;
}
解法2:
刚刚开始给第一个孩子1个糖果,后面的情况可能是:
(1)arr[i]>arr[i-1],则当前i发放糖果数为pre+1;
(2)arr[i]=arr[i-1],则当前i发放糖果数为1;
(3)arr[i]<arr[i-1],则当前i发放糖果数未知,我们会统计下坡的个数,然后在(1)情况中进行处理。
时间复杂度O(n),空间复杂度O(1)。
代码:
public static int candy2(int[] arr){
//pre代表遍历到i时前一个位置i-1分到的糖果数
//count是每次下坡的元素个数,total总糖果数
int pre = 1, count = 0, total = 1;
for(int i = 1; i < arr.length; i++){
if(arr[i] >= arr[i-1]){
//在开始上坡时处理前面的降坡
if(count > 0){
total += count*(count+1)/2; //从右往左开始,1.2.3.....count
//与一个问题,前一个pre(前一个上坡最高点)小于count,则这个pre应该修改为count+1
if(count > pre){
total += count+1-pre;
}
count = 0; //count清零
pre = 1;//之前的pre为前一个升坡最高点,需要更新为1,可以理解为谷底,下一个上波的开始
}
pre = arr[i] == arr[i-1] ? 1 : pre+1; //相等,此时i处小孩分糖果为1,否则前者+1
total += pre;
}else {
count++; //下坡数统计
}
}
//如果遍历到最后一个数时为下坡,count需要处理
if(count > 0){
total += count*(count+1)/2;
if(count > pre){
total += count+1-pre;
}
}
return total;
}