75颜色分类

题目描述

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

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

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

示例:

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

进阶：

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

思路分析

荷兰国旗问题。

设置三个指针left,right,cur，指向左边界、右边界、当前元素，两个指针指向less=left-1,more=right+1，分别表示<num,>num的区域。

• arr[cur]<num，交换cur和less+1上的元素，cur++，less++
• arr[cur]>num，交换cur和more-1上的元素，more–，此时还需检查交换来的元素和num的大小，故不用对cur进行操作
• arr[cur]==num，cur++

swap方法用位运算写程序运行会快一点，因为不用建立tmp

代码实现

    public void sortColors(int[] nums) {
        int less = -1, more = nums.length;
        int cur = 0;
        while (cur < more) {
            //我们知道1放中间，所以以1作为分界数
            if (nums[cur] == 1) {
                cur++;
            } else if (nums[cur] > 1) {
                more--;
                swap(nums, more, cur);
            } else if (nums[cur] < 1) {
                less++;
                swap(nums, less, cur);
                cur++;
            }
        }
    }
 

    public void swap(int[] nums, int a, int b) {
        if(a==b) return;
        nums[a] = nums[a] ^ nums[b];
        nums[b] = nums[b] ^ nums[a];
        nums[a] = nums[a] ^ nums[b];
//        int tmp = nums[a];
//        nums[a] = nums[b];
//        nums[b] = tmp;
    }