计数排序-leetcode-75. 颜色分类

75. 颜色分类

题目描述

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

提示：

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

进阶：

• 你可以不使用代码库中的排序函数来解决这道题吗？
• 你能想出一个仅使用常数空间的一趟扫描算法吗？

解法选择：

1. 快排；
2. 计数排序

快排

直接使用 Arrays.sort() 排序，即可

• 时间复杂度：$O(nlogn)$
• 空间复杂度：$O(logn)$

计数排序

• 分别使用 3 个变量，red、white、blue 统计每种颜色出现的次数；
• 再按照 红色、白色、蓝色顺序排列。

class Solution {
    public void sortColors(int[] nums) {
        int red = 0;
        int white = 0;
        int blue = 0;
        for (int i : nums) { // 统计颜色 次数，即 计数阶段
            if (i == 0) {
                red++;
            } else if (i == 1) {
                white++;
            } else {
                blue++;
            }
        }
        int i = 0;
        // 根据每种颜色的次数，恢复
        for ( ; i < red; i++) {
            nums[i] = 0; // 可以省略，直接从 white 开始
        }
        for ( ; i < white + red; i++) {
            nums[i] = 1;
        }
        for ( ; i < nums.length; i++) {
            nums[i] = 2;
        }
    }
}

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$