leetcode题解75-颜色分类

问题描述

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:

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

解题思路

本题很容易想到的思路是先对数组进行排序，但是题中要求不能排序，想到这里只有3种颜色，即3个数字，我们可以用一趟循环来统计每个数字出现的次数，然后重写数组即可。

代码实现

class Solution {
    public void sortColors(int[] nums) {
        //计算数字0、1、2的数量
		 int nums0=0,nums1=0,nums2=0;
		 for(int i=0;i<nums.length;i++){
			 if(nums[i]==0){
				 nums0++;
			 }else if(nums[i]==1){
				 nums1++;
			 }else{
				 nums2++;
			 }
		 }
		//根据0、1、2的数量来重写数组
		int k=0;
		while(k<nums.length){
			if(k<nums0){
				nums[k++]=0;
			}else if((k>=nums0)&&(k<nums0+nums1)){
				nums[k++]=1;
			}else{
				nums[k++]=2;
			}
		}
    }
}