Leetcode-75 颜色分类

Leetcode-75 颜色分类

荷兰三色旗问题

题目描述

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

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

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶：

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

题目要求原地排序，不可新建数组

1、计数排序
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
两趟扫描

class Solution {
    public void sortColors(int[] nums) {
        int red=0;
        int white=0;
        int blue=0;
        int length=nums.length;
        for(int i=0;i<length;i++)//一趟扫描
        {
            if(nums[i]==0)
                red++;
            else if(nums[i]==1)
                white++;
            else
                blue++;
        }
        //二趟扫描
        for(int i=0;i<red;i++)
            nums[i]=0;
        for(int i=red;i<red+white;i++)
            nums[i]=1;
        for(int i=red+white;i<red+white+blue;i++)
            nums[i]=2;
        return;    
    }
}

2、

class Solution {
    public void sortColors(int[] nums) {
        int right=0;//0的最右边
        int left=nums.length-1;//2的最左边
        int cur=0;//当前元素
        while(cur<=left)//一趟扫描
        {
            if(nums[cur]==0)
            {
                int temp=nums[right];
                nums[right]=nums[cur];
                nums[cur]=temp;
                right++;
                cur++;
            }
            else if(nums[cur]==2)
            {
                int temp=nums[left];
                nums[left]=nums[cur];
                nums[cur]=temp;
                left--;
            }
            else
                cur++;
            } 
    }
}

nums[0-cur-1]之间不会出现2，如果出现2早就被置换到后边去了，所以只有可能是1与nums[cur]置换，这也就是为什么nums[cur]==0时cur++而nums[cur]==2时没有cur++的原因：只能保证cur前面的，但保证不了cur后面的
时间复杂度：