LeetCode - #135 分发糖果

前言

本题由于没有合适答案为以往遗留问题，最近有时间将以往遗留问题一一完善。

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新到 134 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：困难

1. 描述

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

• 每个孩子至少分配到 1 个糖果。
• 相邻两个孩子评分更高的孩子会获得更多的糖果。

请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

2. 示例

示例 1

输入：ratings = [1,0,2]
输出：5
解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。

示例 2

输入：ratings = [1,2,2]
输出：4
解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
     第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。

提示：

• n == ratings.length
• 1 <= n <= 2 * 10^4
• 0 <= ratings[i] <= 2 * 10^4

3. 答案

题解

public class Solution {
   func candy(_ ratings: [Int]) -> Int {
       let size = ratings.count

       if size == 0 {
           return -1
       }
       if size == 1 {
           return 1
       }

       var candys = Array(repeating: 1, count: size)

       for i in 1..<size {
           if ratings[i] > ratings[i - 1] {
               candys[i] = candys[i - 1] + 1
           }
       }
       for i in 1..<size {
           let j = size - i
           if ratings[j - 1] > ratings[j], candys[j - 1] <= candys[j]{
               candys[j - 1] = candys[j] + 1
           }
       }

       var sum = 0
       for i in 0..<size {
           sum += candys[i]
       }
       return sum
   }
}

点击前往 LeetCode 练习

关于我们

Swift社区是由 Swift 爱好者共同维护的公益组织，我们在国内以微信公众号的运营为主，我们会分享以 Swift实战、SwiftUl、Swift基础为核心的技术内容，也整理收集优秀的学习资料。

开启掘金成长之旅！这是我参与「掘金日新计划 · 2 月更文挑战」的第 1 天，点击查看活动详情