每日算法,剑指offer(2)Java篇 分糖果

描述

一群孩子做游戏,现在请你根据游戏得分来发糖果,要求如下:

1. 每个孩子不管得分多少,起码分到一个糖果。

2. 任意两个相邻的孩子之间,得分较多的孩子必须拿多一些糖果。(若相同则无此限制)

给定一个数组 arr 代表得分数组,请返回最少需要多少糖果。

要求: 时间复杂度为 O(n) 空间复杂度为 O(n)

数据范围: 1≤n≤100000 ,1≤ai​≤1000

示例1:

 

   输入:[1,1,2]

   返回值:4

   说明:最优方案1,1,2

示例2:

 

 输入:[1,1,1]

  返回值:3

  说明:最优方案1,1,2
解题思路:使用了算法思想中的贪心算法
import java.util.*;


public class Solution {
       public int candy (int[] arr) {
        if(arr == null || arr.length == 0) return 0 ;
        int[] right = new int[arr.length] ;//存每个小朋友对应的糖果数
        //从前往后遍历,寻找递增的
        for(int i = 0 ; i < arr.length ; i ++) {
            right[i] = 1 ;//先默认为 1
            if(i == 0) continue ;
            //如果是递增,那么在前一个糖果数基础上加一,这个是最少要求的糖果数,不能再小了
            if(arr[i] > arr[i-1]) {
                right[i] = right[i-1] + 1 ;
            }
        }
        //从后往前遍历,寻找递减的
        for(int j = arr.length - 1 ; j > 0 ; j--) {
            if(arr[j-1] > arr[j]) {//考虑递减的情况
                int tmp = right[j] + 1 ;//在后一个的基础上加一,得出这个位置上所需要的最基本的糖果数
                if(tmp > right[j - 1]) {//取正向,和反向求出的该位置的公共最基本糖果数,即,较大值
                    right[j - 1] = tmp ;
                }
            }
        }
        int count = 0 ;
        for(int k : right) {
            count += k ;
        }
        return count ;
    }
}

你可能感兴趣的:(算法,java)