Sort Colors

来源：https://leetcode.com/problems/sort-colors

Given an array with n objects colored red, white or blue, sort them 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.

 

选择排序（Selection sort）是一种简单直观的排序算法。它的工作原理如下。首先在未排序序列中找到最小（大）元素，存放到排序序列的起始位置，然后，再从剩余未排序元素中继续寻找最小（大）元素，然后放到已排序序列的末尾。以此类推，直到所有元素均排序完毕。 --维基百科  https://zh.wikipedia.org/wiki/选择排序

选择排序的复杂度（非本题复杂度）：

时间复杂度，平均O(n^2)，最好O(n^2)，最坏O(n^2)

空间复杂度，辅助存储O(1)

稳定性，不稳定

 

方法：应用选择排序的思想，将最小元素（0）从起始位置开始排，将最大元素（2）从末尾位置开始排，剩下的1自然就排好了

Java

class Solution {
    public void sortColors(int[] nums) {
        int zeroPos = 0, twoPos = nums.length-1, tmp = 0;
        for(int i=0; i<=twoPos; i++) {
            if(nums[i] == 0) {
                tmp = nums[i];
                nums[i] = nums[zeroPos];
                nums[zeroPos] = tmp;
                zeroPos++;
            }
            if(nums[i] == 2 && i < twoPos) {
                tmp = nums[i];
                nums[i] = nums[twoPos];
                nums[twoPos] = tmp;
                twoPos--;
                i--;
            }
        }
    }
}

另一种神奇的方法，本质是记录了0的个数（r），0和1的个数（w），0、1和2的总个数（b）

Python

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        r, w, b = 0, 0, 0
        for i in range(len(nums)):
            if nums[i] == 0:
                nums[b] = 2
                nums[w] = 1
                nums[r] = 0
                b += 1
                w += 1
                r += 1
            elif nums[i] == 1:
                nums[b] = 2
                nums[w] = 1
                b += 1
                w += 1
            else:
                nums[b] = 2
                b += 1