leetcode 75 颜色分类

75. 颜色分类

难度中等397

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

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

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

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶：

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

思路：

比较直观的做法 直接使用 快排方法进行排序
时间复杂度： O(logn)
空间复杂度： O(1)

代码1

class Solution {
    public void sortColors(int[] nums) {
        quickSort(nums, 0, nums.length-1);
    }
    // 来个快排
    public void quickSort(int[] arr, int low, int high){
        int l = low;
        int r = high;

        if(l<r){
            int temp = arr[l];
            while(l<r){
                while(l<r && temp < arr[r]){
                    r--;
                }
                if(l < r){
                    arr[l] = arr[r];
                    l++;
                }

                while(l < r && temp > arr[l]){
                    l++;
                }
                if(l < r){
                    arr[r] = arr[l];
                    r--;
                }
            }
            arr[l] = temp;
            quickSort(arr, low, l-1);  // 不要再把 arr[l] 算进去了 他已经在这次运算中 成为一个 中枢 不再参与运算
            quickSort(arr, l+1, high);
        }
    }
}

思路 2

使用 三个指针： 由于 只有 0 , 1, 2 三个元素【红白蓝】 
p0 指向 0 的最右边界，即每一次 p0 的位置保存 0 的值之后 就进行 p0++
p2 指向 2 的最左边界，即每一次 p2 的位置保存 2 的值之后 就进行 p2--
cur 指向当前元素 进行判断当前元素是 0 还是 1， 这样通过 只考虑 0 和 2 就可以将 1 界定于 0 和 2 之间。
时间复杂度： O(n)
空间复杂度： O(1)

代码2

class Solution {
    public void sortColors(int[] nums) {
        //quickSort(nums, 0, nums.length-1);
        sortColorsDo(nums);
    }
   // 使用 每次cur进行判断是0还是2的方法 
    public void sortColorsDo(int[] nums){
        // p0 指向 0 的最右边界  p2 指向 2的最左边界
        // cur 指向当前元素 不断往后走
        int p0 = 0, cur = 0;
        int p2 = nums.length - 1;
        int tmp;
        while(cur <= p2){
            // 判断 0
            if(nums[cur] == 0){ // 比较元素
                // 和 p0 进行交换
                tmp = nums[p0];
                nums[p0++] = nums[cur];
                nums[cur++] = tmp;
            }
            else if(nums[cur] == 2){
                // 和 p2 进行交换
                tmp = nums[cur];
                nums[cur] = nums[p2];
                nums[p2--] = tmp;
            }else{
                cur++;
            }
        }
    }
   
}