一、题目
描述
一群孩子做游戏,现在请你根据游戏得分来发糖果,要求如下:
- 每个孩子不管得分多少,起码分到一个糖果。
- 任意两个相邻的孩子之间,得分较多的孩子必须拿多一些糖果。(若相同则无此限制)
给定一个数组 arr 代表得分数组,请返回最少需要多少糖果。
要求: 时间复杂度为 O(n) 空间复杂度为 O(n)
数据范围:
1≤n≤100000 ,
1≤a i ≤1000
示例1
输入:
[1,1,2]
返回值:
4
说明:
最优分配方案为1,1,2
示例2
输入:
[1,1,1]
返回值:
3
说明:
最优分配方案是1,1,1
二、思路
- 因为是求最小糖果数,先初始化每个人给一颗糖
- 从左到右遍历,如果分数大于前一个,则糖果个数=前一个的糖果数+1
- 从右到左遍历,如果分数大于后一个且拥有糖果数小于后一个,给它的糖果数=后一个的糖果数+1
三、代码
import java.util.*;public class Solution {/*** 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可** pick candy* @param arr int整型一维数组 the array* @return int整型*/public int candy (int[] arr) {// write code hereArrayList<Integer> nums = new ArrayList<>();for(int i:arr){nums.add(i);}//1、将初始分数为1List<Integer> candys = new ArrayList<>(Collections.nCopies(nums.size(), 1));//2、从左至右判断 如果分数大于前一个 则糖果个数置为前一个的糖果个数+1for(int i = 1; i < nums.size() ; i++){if(nums.get(i)>nums.get(i-1)){candys.set(i,candys.get(i-1)+1);}else {candys.set(i,1);}}//3、从右至左判断 如果分数大于前一个 则糖果个数置为前一个的糖果个数+1for(int i = nums.size()-2; i >=0 ; i--){if(nums.get(i)>nums.get(i+1) && candys.get(i)<=candys.get(i+1)){candys.set(i,candys.get(i+1)+1);}}//4、收集糖果个数int sum=0;for(int i : candys){sum+=i;}return sum;}
}
四、题目来源
牛客,BM95 分糖果问题