分糖果问题（java实现）

一、题目

描述
一群孩子做游戏，现在请你根据游戏得分来发糖果，要求如下：

1. 每个孩子不管得分多少，起码分到一个糖果。
2. 任意两个相邻的孩子之间，得分较多的孩子必须拿多一些糖果。(若相同则无此限制)

给定一个数组 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. 因为是求最小糖果数，先初始化每个人给一颗糖
2. 从左到右遍历，如果分数大于前一个，则糖果个数=前一个的糖果数+1
3. 从右到左遍历，如果分数大于后一个且拥有糖果数小于后一个，给它的糖果数=后一个的糖果数+1

三、代码

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * pick candy
     * @param arr int整型一维数组 the array
     * @return int整型
     */
    public int candy (int[] arr) {
        // write code here
        ArrayList<Integer> nums = new ArrayList<>();
        for(int i:arr){
            nums.add(i);
        }
        //1、将初始分数为1
        List<Integer> candys = new ArrayList<>(Collections.nCopies(nums.size(), 1));

        //2、从左至右判断 如果分数大于前一个 则糖果个数置为前一个的糖果个数+1
        for(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、从右至左判断 如果分数大于前一个 则糖果个数置为前一个的糖果个数+1
        for(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 分糖果问题