Leetcode刷题笔记——颜色分类

题目描述：

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

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

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

示例:

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

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

解题思路：

　　一次遍历完成排序，那么就要在遍历的过程中完成移动。思路是，记录下已经碰到的0的数量nums0和2的数量nums2，初始都为0。遍历过程中，碰到数字0，就往前面挪，具体来说就是当前位置i上的数与下一个0的放置位上的数进行交换，下一个0的放置位为nums[nums0]；同理，碰到2的话，则交换i位置和nums[nums.length-1-nums2]位置上的数；当碰到1时，不做交换，i向前走1位。

　　需要注意的一点是，当碰到数组起始位为0时，交换的两个位置相同，而交换完了nums0加一，i却不变，这样会造成错误，此当i

代码实现：

/***
     * 解题思路：一次遍历，碰到0往前挪，碰到2往后挪，碰到1不动，但指针加1
     * @param nums
     */
    public static void sortColors(int[] nums){
        if(nums == null || nums.length == 0)
            return;
        //已碰到的数字0的个数，下一个0的放置位为nums[nums0]
        int nums0 = 0;
        //已碰到的数字2的个数，下一个2的放置位为nums[nums.length-1-nums2]
        int nums2 = 0;

        int i = 0;
        while (inums2){
            if(nums[i] == 0){
                swap(nums, nums0, i);
                nums0++;
                //如果指针的位置和下一个0的放置位重合，那么0的个数增加1，指针也要向前走一位。
                if(i<nums0)
                    i++;
            }
            else if(nums[i] == 1){
                i++;
            }
            else{
                swap(nums, nums.length-1-nums2, i);
                nums2++;
            }
        }
    }

    public static void swap(int[] nums, int i, int j){
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }

另外，这道题是我自己想出来的，看了官方题解说这个是“荷兰国旗问题”，而且解题思路和我想的基本一致，哈哈哈我太聪明了！

转载于:https://www.cnblogs.com/yingying7/p/11320786.html