LeetCode 75 —— SortColors

题目

Given an array with n objects colored red, white or blue, sort them in-place so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note: You are not suppose to use the library's sort function for this problem.

Example:

Input: [2,0,2,1,1,0]
Output: [0,0,1,1,2,2]

Follow up:

• A rather straight forward solution is a two-pass algorithm using counting sort.
  First, iterate the array counting number of 0's, 1's, and 2's, then overwrite array with total number of 0's, then 1's and followed by 2's.
• Could you come up with a one-pass algorithm using only constant space?

 

---

 

题目解读

给定包含N个元素的数组，包含红、白、蓝三种颜色，按照红白蓝的顺序排序使相同的颜色毗邻，且在数组内排序。

我们使用0、1、2分别代表红、白、蓝三种颜色。

注意：不能使用库中的排序函数。

进阶要求：一个比较直接的解决方式是通过两次遍历的算法。第一次遍历数组计算0、1、2的个数，然后第二次遍历时，用对应个数的0、1、2覆盖数组。是否可以用一次遍历且O(1)的空间复杂度来解决问题呢？

 

解题思路

简单的思路已经在进阶要求中提到了，现在我们直接尝试一次遍历解决排序问题。通常看到“常数空间”、“一次遍历”，需要纳入考虑范围的就是"two pointer"，但是这里有三种颜色，可能单纯的两个指针解决不了问题，那么三个可不可以呢？当然是可以的，利用0、1、2顺序排列的特性，所有的0最终的位置在数组的最前边，所有的2一定在数组最后边，那么我们可以用一个指针遍历数组，其余两个指针用于填充0和2。

 

大体的思路就是这些，但是要注意迭代指针与0或2指针交换后是否仍需要交换。

 

---

 

解题代码

我的写法：

class Solution {
    public void sortColors(int[] nums) {
        int numLen = nums.length;
        int red = 0, blue = numLen - 1;

        for (int i = 0; i <= blue;) {
            if (nums[i] == 0 && i != red) {
                nums[i] = nums[red];
                nums[red++] = 0;
            } else if (nums[i] == 2) {
                nums[i] = nums[blue];
                nums[blue--] = 2;
            } else {
                ++i;
            }
        }
    }
}

 

---

 

如有错误，欢迎指摘。欢迎通过左上角的“向TA提问”按钮问我问题，我将竭力解答你的疑惑。