LeetCode 75. 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.
• 地址

问题分析

• 荷兰国旗问题。将0，1，2分成三个区域。
• 解法1：计数
  
  • 利用计数排序来统计数组中0，1，2的个数，然后重新根数三者个数设置数组
  • 时间复杂度：O(N) ,需要两次遍历数组。
• 解法2：利用三路快排中的 Partition 过程
  
  • 三路快排适用于数组中含有大量重复元素的排序。一次Partiton过程，可以将数组分为 等于pivot区，小于pivot区，大于pivot区。
  • 那么对于该题，如果pivot为1的话，便可以将数组分为 0区域，1区域，2区域。即为所求
  • 时间复杂度：O(N) ,只需1次遍历数组。然而该法其实比计数排序在运行时间上近似，因为Partition过程中含有大量的交换操作。

经验教训

• 三路快排中的 Partition 过程可将数组分为三个区

代码实现

• 计数（尽可能的精简了）

    public void sortColors(int[] nums) {
        if (nums == null || nums.length == 0) {
            return;
        }
        int[] bucket = new int[3];
        for (int i = 0; i < nums.length; i++) {
            ++bucket[nums[i]];
        }
        for (int bucketIndex = 0, i = 0; bucketIndex < 3; ++bucketIndex) {
            for (int count = 0; count < bucket[bucketIndex]; ++count) {
                nums[i++] = bucketIndex;
            }
        }
    }

• 利用三路快排中的 Partition 过程

    public void sortColors(int[] nums) {
        if (nums == null || nums.length == 0) {
            return;
        }
        int zeroRight = -1;
        int twoLeft = nums.length;
        int pivot = 1;
        int i = 0;
        while (i < twoLeft) {
            if (nums[i] < pivot) {
                swap(nums, i++, ++zeroRight);
            }else if (nums[i] > pivot) {
                swap(nums, i, --twoLeft);
            }else {
                i++;
            }
        }
    }

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