leetcode--排序4（75/medium/java）

leetcode

按照模块刷题，点击进入使用的刷题目录

排序

4.颜色分类

题目描述：
颜色分类

解题思路：
法一：设置长度为3的数组，将0，1，2的个数分别存放在数组中。
法二(改进)：设置首尾两个指针，同时设置一个指针遍历整个数组。若元素值为0，则放在数组首，若为2则放在数组尾。0和2放在正确的位置后，1也一定在正确的位置。

代码:
法一：

class Solution {
    public void sortColors(int[] nums) {
        int[] count=new int[3];
        count[0]=count[1]=count[2]=0;
        for(int i=0;i<nums.length;++i){
            int j=nums[i];
            count[j]+=1;
        }
        int a=0;
        for(int i=0;i<3;++i){
            for(int j=0;j<count[i];++j){
                nums[a++]=i;
            }
        }
    }
}

法二：

class Solution {
    public void sortColors(int[] nums) {
        int zero = -1, one = 0, two = nums.length;
        while(one < two){
            if(nums[one] == 0){
                swap(nums, ++zero, one++);
            }else if(nums[one] == 2){
                swap(nums, --two, one);
            }else{
                ++one;
            }
        }
    }
    private void swap(int[]nums, int i, int j){
        int t = nums[i];
        nums[i] = nums[j];
        nums[j] = t;
    }
}

注:
当遍历整个数组的指针和数组尾元素交换时，指针无需自增1，因为交换过来的元素可能是0。