75-颜色分类-爱代码爱编程

题目

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

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

示例 1：

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

示例 2：

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

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

题解

class Solution {
    public void sortColors(int[] nums) {
        int len=nums.length;
        //快速排序
        //[0,zeros),[zeros,two),[two,len-1]一开始的三个指针要保证区间为0
        //最终的终止条件是i与two相同
        if(len<2){
            return;
        }
        int zeros=0;
        int i=0;
        int two=len;
        while(i<two){
            if(nums[i]==0){
                swap(nums,i,zeros);
                i++;
                zeros++;
            }else if(nums[i]==1){
                i++;
            }else{
                two--;
                swap(nums,i,two);
                //此处不需要i++，two指针先前移，再交换，此时two指向的是一个新的小数，交换过后i指向的也是一个新的小数，需要继续判断，所以不需要移动i
            }
        }
    }
    private void swap(int[] nums,int index1,int index2){
        int temp=nums[index1];
        nums[index1]=nums[index2];
        nums[index2]=temp;
    }
}

笔记：

1、注意交换前后指针移动的情况。