LeetCode 135:分糖果

LeetCode 135:分糖果

问题本质与核心挑战

给定孩子的评分数组,需满足 "每个孩子至少1颗糖果,相邻评分高的孩子糖果更多",求最少糖果总数。核心挑战:

  • 相邻约束是双向的(左→右和右→左都需满足),直接枚举无法高效解决;
  • 需通过 两次贪心遍历(左→右、右→左)分别处理单向约束,再合并结果。

核心思路:双向贪心 + 合并约束

1. 单向约束处理(左→右遍历)
  • 定义数组 leftleft[i] 表示 从左到右遍历 时,第 i 个孩子应得的最少糖果(仅满足"左边约束":若 ratings[i] > ratings[i-1],则 left[i] = left[i-1]+1;否则为 1)。
2. 单向约束处理(右→左遍历)
  • 定义数组 rightright[i] 表示 从右到左遍历 时,第 i 个孩子应得的最少糖果(仅满足"右边约束":若 ratings[i] > ratings[i+1],则 right[i] = right[i+1]+1;否则为 1)。
3. 合并双向约束
  • 每个孩子最终的糖果数为 max(left[i], right[i])(同时满足左右约束),求和即为答案。

算法步骤详解(以示例 ratings = [1,0,2] 为例)

步骤 1:左→右遍历,处理左边约束
孩子索引 i 评分 ratings[i] 与左边比较(ratings[i] > ratings[i-1] left[i] 计算 left 数组
0 1 -(无左边孩子) 初始为 1 [1]
1 0 0 > 1?否 设为 1 [1,1]
2 2 2 > 0?是 left[1]+1 = 2 [1,1,2]
步骤 2:右→左遍历,处理右边约束
孩子索引 i 评分 ratings[i] 与右边比较(ratings[i] > ratings[i+1] right[i] 计算 right 数组
2 2 -(无右边孩子) 初始为 1 [1]
1 0 0 > 2?否 设为 1 [1,1]
0 1 1 > 0?是 right[1]+1 = 2 [2,1,1]
步骤 3:合并约束,计算总和
孩子索引 i left[i] right[i] max(left[i], right[i]) 贡献糖果数
0 1 2 2 2
1 1 1 1 1
2 2 1 2 2
总和 - - - 2+1+2=5

完整代码(Java)

java 复制代码
class Solution {
    public int candy(int[] ratings) {
        int n = ratings.length;
        if (n == 0) return 0; // 边界处理(题目保证n≥1,可省略)
        
        // 1. 左→右遍历,处理左边约束
        int[] left = new int[n];
        left[0] = 1; // 第一个孩子至少1颗
        for (int i = 1; i < n; i++) {
            if (ratings[i] > ratings[i-1]) {
                left[i] = left[i-1] + 1;
            } else {
                left[i] = 1;
            }
        }
        
        // 2. 右→左遍历,处理右边约束
        int[] right = new int[n];
        right[n-1] = 1; // 最后一个孩子至少1颗
        for (int i = n-2; i >= 0; i--) {
            if (ratings[i] > ratings[i+1]) {
                right[i] = right[i+1] + 1;
            } else {
                right[i] = 1;
            }
        }
        
        // 3. 合并双向约束,计算总和
        int total = 0;
        for (int i = 0; i < n; i++) {
            total += Math.max(left[i], right[i]);
        }
        return total;
    }
}

关键逻辑解析

  1. 左→右遍历 :确保 "右边孩子评分更高时,糖果数比左边多" ,但无法处理"左边孩子评分更高,右边需更多"的情况(如 [5,4,3,2,1],左遍历后所有 left[i]=1,需右遍历修正)。
  2. 右→左遍历 :确保 "左边孩子评分更高时,糖果数比右边多",与左遍历互补。
  3. 合并约束:取两者最大值,保证同时满足左右双向约束,且糖果数最少(贪心思想:仅在必要时增加糖果)。

复杂度分析

  • 时间复杂度O(n)(两次遍历数组,每次 O(n),合并遍历 O(n))。
  • 空间复杂度O(n)(存储 leftright 数组,可优化为 O(1),但代码可读性降低)。

该方法通过 两次贪心遍历拆分双向约束,将复杂的双向问题转化为两个单向问题,再合并求解,确保了效率和可读性的平衡,是处理此类"双向约束"问题的经典思路。

相关推荐
Raink老师9 小时前
【AI面试临阵磨枪-69】如何设计一个支持百万级工具的 Agent 系统?如何快速路由与选择工具?
人工智能·面试·职场和发展
云泽8089 小时前
笔试算法 -位运算篇(二):从唯一字符到消失数字
c++·算法·位运算
ʚ希希ɞ ྀ9 小时前
不同路径|| -- dp
算法
IT 行者10 小时前
SimHash 与 MinHash:相似性计算的双子星算法
算法·hash·比对
智者知已应修善业10 小时前
【51单片机8位数码管动态显示日期小数点风格】2023-11-13
c++·经验分享·笔记·算法·51单片机
智者知已应修善业10 小时前
【51单片机有三个LED 分别第一个灯闪三下 再到第二个灯又闪三下 再到第三个灯又闪三下 就这样循环程序】2023-11-16
c++·经验分享·笔记·算法·51单片机
小L~~~12 小时前
基于贪心策略的混合遗传算法求解01背包问题
python·算法
洛水水12 小时前
【力扣100题】53.最长回文子串
算法·leetcode·职场和发展
jieyucx12 小时前
Go 语言 sort 包详解:从基础排序到自定义排序(含底层原理+零基础看懂)
算法·golang·排序算法·sort