leetcode 75. 颜色分类(简单模拟)

题目描述

题目链接：leetcode 75. 颜色分类
给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库的sort函数的情况下解决这个问题。
示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]
示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

提示：

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

解题思路

抓住数组中只有0，1，2这三个数的特点，用h[i]表示数i（i=0、1、2）出现的次数，再将0、1、2这三个数依据各自的个数添加到nums中即可。

Java代码

class Solution {
    public void sortColors(int[] nums) {
          int []h = new int[3];
          for(int num : nums){
              h[num]++;
          }
          int p = 0,color = 0;
          for(int x : h){
             while(x > 0){
                  nums[p++] = color ;
                  x--;
             }
             color++; 
          }
    }
}