《快排模板》75. 颜色分类《leetcode》

75. 颜色分类

难度中等829

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

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

 

示例 1：

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

示例 2：

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

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

 

提示：

• n == nums.length
• 1 <= n <= 300
• nums[i] 为 0、1 或 2

代码：快排模板  时间复杂度O（nlogn）

class Solution {
    public static void quickSort(int l,int r,int []nums)
    {
        if(l>=r) return;
        int i=l;
        int j=r;
        int flag=nums[i];
        while(i<j)
        {
            while(nums[j]>flag&&i<j)
                j--;
            if(i<j)
                nums[i++]=nums[j];
            while(nums[i]<flag&&i<j)
                i++;
            if(i<j)
                nums[j--]=nums[i];
        }
        nums[i]=flag;
        //System.out.println(l+" "+r);
        quickSort(l,i-1,nums);
        quickSort(i+1,r,nums);
        return;
    }

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