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

灰太狼 2023-06-14 05:30 2阅读 0赞

### 题目描述 ###

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

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

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

**示例:**

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

**进阶：**

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

### 题目解析 ###

可以使用快排的方法解决，当然，也可以试着用重写数组的方法，统计各个元素出现的个数，在重写一遍数组也是可以实现的，这样做的好处是时间复杂度低。

*  遍历一次数组，统计0,1,2出现的次数
 *  重写一遍数组，先写若干个0，再写若干个1，最后是若干个2

### 实现代码 ###

class Solution {
        
        public void sortColors(int[] nums) {
        
            if(nums==null || nums.length==0) {
        
    			return;
    		}
    		int zero = 0;
    		int one = 0;
    		int two = 0;
    		int j = 0;
    		//遍历一次数组，统计0,1,2出现的次数
    		for(int i=0;i<nums.length;i++) {
        
    			if(nums[i]==0) {
        
    				++zero;
    			} else if(nums[i]==1) {
        
    				++one;
    			} else {
        
    				++two;
    			}
    		}
    		//重写一遍数组，先写若干个0，再写若干个1，最后是若干个2
    		//重写完之后，整个数组就是有序的了
    		while(j<zero) {
        
    			nums[j] = 0;
    			++j;
    		}
    		while(j<zero+one) {
        
    			nums[j] = 1;
    			++j;
    		}
    		while(j<zero+one+two) {
        
    			nums[j] = 2;
    			++j;
    		}
        }
    }

![在这里插入图片描述][20191118210157213.bmp]

[20191118210157213.bmp]: https://img-blog.csdnimg.cn/20191118210157213.bmp