[LeetCode 75. 颜色分类] ｜刷题打卡

一、题目描述：

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。 此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例 1：

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

输出：[0,0,1,1,2,2]

引自：leetcode-cn.com/problems/so…

题目链接：75. 颜色分类

二、思路分析：

1. 遍历颜色数组，并且记录下0、1、2的数量；
2. 按照记录的0、1、2的数量，依次给数组的元素重新进行复制；
3. 注意重新赋值1时的边界。

三、AC 代码：

class Solution {
    func sortColors(_ nums: inout [Int]) {
        if (nums.count < 2) {
            return;
        }
        var zeroCount : Int = 0;
        var oneCount : Int = 0;
        var twoCount : Int = 0;
        for value : Int in nums {
            switch value {
                case 0:
                    zeroCount += 1;
                    break;
                case 1:
                    oneCount += 1;
                    break;
                case 2:
                    twoCount += 1;
                    break;
                default:
                    break;
            }
        }
        
        var i : Int = 0;
        while (i < nums.count) {
            if (i < zeroCount) {
                nums[i] = 0;
            } else if (i < oneCount+zeroCount) {
                nums[i] = 1;
            } else {
                nums[i] = 2;
            }
            i += 1;
        }
    }
}

四、参考学习网址

本文正在参与「掘金 3 月闯关活动」, 点击查看 活动详情