leetcode75.颜色分类

题目描述

---

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

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

注意

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

示例：

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

进阶：

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

解题思路：

---

根据进阶提示，这题这里有两种解法，都是O(n)的时间复杂度，和常数的空间。

1. 第一种就是最直观的，先迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。需要两趟循环。 2. 根据三路快排的思想，将0放在数组前面，2放在数组后面。只需要一趟循环。

第一种解法：

	public void sortColors(int[] nums) {
        if (nums.length == 1) return;
        int[] num = new int[3]; //记录0,1,2的个数
        for (int i = 0; i < nums.length; i ++)
            num[nums[i]] ++;

        int index = 0;

        for (int i = 0; i < num.length; i ++) {
            int n = num[i];
            while (n > 0) {
                nums[index] = i;
                ++ index;
                -- n;
            }
        }
    }

第二种解法：

	
    public void sortColors(int[] nums) {
        if (nums.length == 1) return;
        int indexStart = 0, indexEnd = nums.length - 1; //indexStart记录当前放0的位置，indexEnd记录放2的位置

        for (int i = 0; i <= indexEnd; i ++ ) {
            if (nums[i] == 0) {
                nums[i] = nums[i] ^ nums[indexStart];
                nums[indexStart] = nums[i] ^ nums[indexStart];
                nums[i] = nums[i] ^ nums[indexStart];
                ++ indexStart;
            }
            else if (nums[i] == 2 && i != indexEnd) {
                nums[i] = nums[i] ^ nums[indexEnd];
                nums[indexEnd] = nums[i] ^ nums[indexEnd];;
                nums[i] = nums[i] ^ nums[indexEnd];
                -- indexEnd;
                -- i; //因为换过来的数字并为被遍历过，所以需要对它进行判断
            }
        }
    }

贴上自己的博客地址：http://rikin04.club/2019/04/20/Leetcode75. 颜色分类/