每日一道算法题（11.21）

• 题目来自力扣
• 文章只记录个人解题思路，并非最优
• 每周一至周五每日一题，周末不更新

75. 颜色分类

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

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:

不能使用代码库中的排序函数来解决这道题。

实例

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶

• 一个直观的解决方案是使用计数排序的两趟扫描算法。 首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
• 你能想出一个仅使用常数空间的一趟扫描算法吗？

链接：leetcode-cn.com/problems/so…

解题

1.第一反应sort，简单

var sortColors = function(nums) {
    Object.assign(nums, nums.sort((a, b) => a - b)) 
};

但是题目注明了不能使用代码库中的排序函数来解决这道题。 所以选择了另一种：
2. 0，1，2 排序。一次遍历，如果是0，则移动到表头，如果是2，则移动到表尾，不用考虑1

var sortColors = function(nums) {
    for (let i = j = 0; i < nums.length && j < nums.length; i++, j++) {
        if (nums[i] === 0) {
            nums.unshift(nums.splice(i, 1))
        } else if (nums[i] === 2) {
            nums.push(nums.splice(i, 1))
            i --
        } else {}
    }
};