颜色分类问题 快速排序day02

颜色分类问题 leetcode 75

这道题我开始当成排序题做了，因为要求是进行原地排序，所以我使用的是插入排序

public static void sortColors(int[] nums) {
//        使用插入排序
        if (nums.length<2||nums==null) return;
        for (int i = 1; i <nums.length ; i++) {
            for (int j = i-1; j >-1 ; j--) {
//                && nums[j]>nums[j+1]
                if (nums[j]>nums[j+1]) {
                    int middle=nums[j];
                    nums[j]=nums[j+1];
                    nums[j+1]=middle;
                }
            }
        }
    }

使用插入排序的时间复杂度是O($n^2$)
想到了荷兰国旗问题，所以用了三指针法优化代码

优化代码

public static void sortColors(int[] nums) {
        if (nums==null||nums.length<2)return;
        int middle=1;//设置中间的partition，小于它的放左边，大于它的放右边
        int l=0;
        int r=nums.length-1;
        int cur=0;
        while (cur<=r){//注意这里停止循环的条件是当前指针<=右边界
            if (nums[cur]<middle){
                swap(nums,cur,l);
                cur++;
                l++;
            }
            else if (nums[cur]>middle){
                swap(nums,cur,r);
                r--;//因为从有边界换过来的都是新的没有比较过的值，所以cur不能++
            }
            else cur++;
        }
    }
    public static void swap(int[] nums,int i,int j){
        int temp=nums[i];
        nums[i]=nums[j];
        nums[j]=temp;
    }

复习快速排序

1. 首先在数组中找一个基准数，方便起见找第一个数为基准数
2. 将这个数组中所有比基准数大的数放在右边，比基准数小的数放在左边

举例

列表为3 1 2 5 4 6 9 7 10 8

我们以数字6作为基准数，i向右，j向左，当i遇到一个比6大的数先停下来，j遇到一个比6大的数也停下来
交换这两个数的位置

继续移动，此时i在9停下，j在7停下，交换这两个数

继续移动，i和j相遇，把基准数和此时的i（j）交换，第一次操作结束，此使基准数左边比这个数小，基准数右边比基准数大。我们发下接下来只需要对左右两边重复上述操作即可完成任务（递归过程）

public static void sortColors(int[] nums){
        int low=0;
        int high=nums.length-1;
        quicksort(nums,low,high);

    }
    public static void quicksort(int[] nums,int l,int r){
        if (l>r)return;//递归出口
        int i,j,base;
        i=l;
        j=r;
        base=nums[i];
        while (i<j){
            while (i<j&&nums[j]>=base)//先找右边（必须），从右边找一个比基准大的
                j--;
            while (i<j&&nums[i]<=base)//再找左边，从左边找一个比基准小的
                i++;

            if (i<j){//如果满足左指针小于右指针，交换
                int middle=nums[i];
                nums[i]=nums[j];
                nums[j]=middle;
            }
        }
        int temp=nums[i];//基准数与i（j）所在的值交换，此时基准数左边比它小，右边比它大
        nums[i]=base;
        nums[l]=temp;
        quicksort(nums,l,j-1);//左递归
        
        quicksort(nums,j+1,r);//右递归
    }