Leetcode 颜色分类

颜色分类

题目描述：

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

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

提示：

   n == nums.length

   1 <= n <= 300

   nums[i] 为 0、1 或 2

进阶：

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

题目链接

class Solution {
    public void sortColors(int[] nums) {
        // 收集每个颜色所含有的个数
        int[] color = new int[3];// 常数空间
        int len = nums.length;
        for(int i = 0 ; i<len ; i++){
            color[nums[i]]++;
        }
        int j = 0;
        for(int i = 0 ; i<len ;j++){
            if(color[j] != 0){
                while(color[j] != 0){
                    nums[i] = j;
                    color[j]--;
                    i++; // 这里改变了最外层循环的i
                }
            }
        }
    }
}

该题总共遍历两次。对于第一次遍历：由于颜色只有三种，此时将颜色收集到color数组中；对于第二次遍历：将color数组中的颜色按照顺序分布到数组中，详细请看代码。