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、注意交换前后指针移动的情况。

上一篇:Leetcode 350:Intersection of Two Arrays II


下一篇:AT2675 [AGC018F] Two Trees