Leetcode 135. 分发糖果（问题分解）

ooold_six

于 2024-06-19 14:50:38 发布

阅读量464

点赞数 2

分类专栏： LeetCode 文章标签： leetcode 算法数据结构 java

本文链接：https://blog.csdn.net/ooold_six/article/details/139802643

版权

LeetCode 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

Leetcode 135. 分发糖果

根据描述，可知更多糖果发生在相邻两个孩子的rating更高者中，对于一个孩子来说，左右两侧都视为相邻，即ratings[ i ] > ratings[ i-1 ] 或 ratings[ i ] > ratings[ i+1 ]都会令糖果增加

由此则将问题分解成两个方面，分别考虑ratings大于左侧和ratings大于右侧两种情况
使用nums[ ] 记录每个孩子的糖果数，这个数组初始化为1

从左向右考虑ratings大于左侧，即ratings[ i ] > ratings[ i-1 ]情况，此时令 i 处的糖果数nums[ i ]直接赋值为nums[ i-1 ] + 1即可满足条件

接下来从右向左考虑ratings大于右侧情况，即ratings[ i ] > ratings[ i+1 ]情况
注意的是，此时nums并非全1的初始化状态，而是已经计算过的数组，若ratings[ i ] > ratings[ i+1 ]发生，而此时nums[ i ] > nums[ i+1 ]已经满足，则无需再次分配糖果，否则同理左遍历为nums[ i ]赋值

如ratings{5，10，7}，左遍历后nums{1，2，1}，此时进行右遍历则无需为nums[ 1 ]分配更多糖果

class Solution {
    public int candy(int[] ratings) {
        int n = ratings.length;
        int nums[] = new int [n];
        Arrays.fill(nums, 1);
        for(int i = 1; i < n; i ++){
            if(ratings[i] > ratings[i-1])
                nums[i] = nums[i-1] + 1;
        }
        for(int i = n-2; i >= 0; i --){
        	// 特判
            if(ratings[i] > ratings[i+1] && nums[i] <= nums[i+1])
                nums[i] = nums[i+1] + 1;
        }
        int sum = 0;
        for(int i : nums){
            sum += i;
        }
        return sum;
    }
}