一起养成写作习惯！这是我参加「掘金日新计划 4 月更文应战」的第21天，点击查看活动概况。

前语

本题为 LeetCode 前 100 高频题

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

LeetCode 算法到现在咱们现已更新了 74 期，咱们会保持更新时间和进展（周一、周三、周五早上 9:00 发布），每期的内容不多，咱们期望咱们可以在上班路上阅览，长久积累会有很大提高。

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

难度水平：中等

1. 描绘

给定一个包含赤色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并依照赤色、白色、蓝色顺序排列。

咱们运用整数 0、 1 和 2 分别表示赤色、白色和蓝色。

必须在不运用库的 sort 函数的情况下处理这个问题。

2. 示例

示例 1

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2

输入：nums = [2,0,1]
输出：[0,1,2]

约束条件：

  • n == nums.length
  • 1 <= n <= 300
  • nums[i] 为 0、1 或 2

3. 答案

class SortColors {
    func sortColors(_ nums: inout [Int]) {
        var red = 0, blue = nums.count - 1, i = 0
        while i <= blue {
            if nums[i] == 0 {
                _swap(&nums, i, red)
                red += 1
                i += 1
            } else if nums[i] == 1 {
                i += 1
            } else {
                _swap(&nums, i, blue)
                blue -= 1
            }
        }
    }
    fileprivate func _swap<T>(_ nums: inout [T], _ p: Int, _ q: Int) {
        (nums[p], nums[q]) = (nums[q], nums[p])
    }
}
  • 首要思想：桶排序。
  • 时间复杂度： O(n)
  • 空间复杂度： O(1)

该算法题解的库房：LeetCode-Swift

点击前往 LeetCode 练习

关于咱们

咱们是由 Swift 爱好者共同维护，咱们会共享以 Swift 实战、SwiftUI、Swift 基础为中心的技能内容，也收拾收集优秀的学习材料。