【75】颜色分类（已优化）

题目（难度：中等）：

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

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

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

示例:

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

代码思想：

归并排序

代码实现：

public void sortColors(int[] nums) {
        if (nums == null || nums.length < 2) {
            return;
        }
        mergeSort(nums, 0, nums.length - 1);
    }

    public static void mergeSort(int[] array, int l, int r) {
        if (l == r) return;
        int mid = l + ((r - l) >> 1);
        mergeSort(array, l, mid);
        mergeSort(array, mid + 1, r);
        merge(array, l, mid, r);
    }

    public static void merge(int[] array, int l, int mid, int r) {
        int[] help = new int[r - l + 1];
        int i = 0, p1 = l, p2 = mid + 1;
        while (p1 <= mid && p2 <= r) {
            help[i++] = array[p1] < array[p2] ? array[p1++] : array[p2++];
        }
        while (p1 <= mid) {
            help[i++] = array[p1++];
        }
        while (p2 <= r) {
            help[i++] = array[p2++];
        }
        for (i = 0; i < help.length; i++) {
            array[l + i] = help[i];
        }
    }

测试结果：

运行结果：

算法分析：

时间复杂度O(n*lgn)，空间复杂度O(n)