【LeetCode刷题】75. 颜色分类

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

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

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

示例:

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

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

===============================================================================================================================================================

分析：

法一：按照题目给出的思路，遍历数组，统计0、1、2出现的次数，再排序（赋值）

class Solution {
    public void sortColors(int[] nums) {
        int red = 0,white = 0,blue = 0;
        for(int i=0;i<nums.length;i++){
            switch(nums[i]){
                case 0:red++;break;
                case 1:white++;break;
                case 2:blue++;break;
            }
        }
    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<nums.length;i++){
        nums[i]=2;
    }

  }
}

法二：按照官方给出的解析，定义一个指针p，指向头部，第一次遍历数组，找出数组中数值为0的值，与nums[p]交换，p后移；第二次遍历数组，找出数组中数值为0的值，与nums[p]交换，p后移。

class Solution {
    public void sortColors(int[] nums) {
        int len = nums.length;
        int ptr = 0;
        for(int i =0;i<len;i++){
            if(nums[i]==0){
                int temp = nums[i];
                nums[i] = nums[ptr];
                nums[ptr] = temp;
                ptr++;
            }
        }

         for(int i =0;i<len;i++){
            if(nums[i]==1){
                int temp = nums[i];
                nums[i] = nums[ptr];
                nums[ptr] = temp;
                ptr++;
            }
        }

    }
}

有了法二这种交换之后，我就想到定义两个指针，一个p0在头部，指向0的位置，一个p2在尾部，指向2的位置；然后遍历数组，当数值为0时，交换num[i]和num[p0]；同样的，当数值为2时，交换num[i]和num[p2]。这里注意不要越界就行了。才发现官方第三中方法就是这样。。。。。。

class Solution {
    public void sortColors(int[] nums) {
        int len = nums.length;
        int p0 = 0,p2 = len-1;
        for(int i=0;i<=p2;i++){
            while (i <= p2 && nums[i] == 2) {
                int temp = nums[i];
                nums[i] = nums[p2];
                nums[p2] = temp;
                p2--;
            }
            if (nums[i] == 0) {
                int temp = nums[i];
                nums[i] = nums[p0];
                nums[p0] = temp;
                p0++;
            }
    }
}
}

参考：https://leetcode-cn.com/problems/sort-colors