利用移除元素思想解决快排

发布时间 2023-08-10 13:03:46作者: xiuer211

在学到快排的时候感觉快排很难,现在捋顺之后,思路廓然开朗了。
先来看一道leetcode简单题目
27. 移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:

输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,
而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。

这里就不过多解释了,直接上代码

 public int removeElement(int[] nums, int val) {
        int fast=0;
        int slow=0;
        while (fast<nums.length){
            if(nums[fast]!=val){
                nums[slow]=nums[fast];
                slow++;
            }
                fast++;
        }
        return slow;
    }

当初我看到的时候是一脸懵的,为啥是!=,为啥要nums[slow]=nums[fast];
现在就来解答:
题目需要找到值不是val的所有元素,我们就想办法把值不是val的元素往前移,我们需要两个指针,来记录什么时候把值不是val的元素往前移。一个是fast,一个是slow来保存数组索引
那么如何做到把值不是val的元素往前移,答案很简单:fast负责遍历,slow符合记录值是val的元素的下标。当fast找到符合条件的就把fast索引的元素保存到slow索引上。

但是为什么slow要记录记录值是val的元素的下标呢?因为我们要把不是val的元素保存到slow里边,如果slow下标不是我们想要的,自然就需要把slow覆盖掉。

但是这跟快排有什么关系呢?
快排思想是:给定一个中心点pivotpivot放到合适的位置,自然就是左边都比pivot小,右边都比pivot大。

那么怎么找到左边都比pivot小的下标呢,上面的代码已经说明了一切,只需要定义两个指针:一个是fast,一个是slow来保存数组索引

fast负责遍历,slow符合记录比pivot大的元素。当fast找到符合条件的就把fastslow两个索引对应的元素进行交换,因为这是排序。

fast找到high的时候,此时的slow自然就是pivot的应该呆在的地方,再交换一次。这样我们就完成了一次,剩下的就交给递归处理。

    public static int[] quickSort(int[]arr,int low,int high){
        if (low<high){
            int postion = quickSortHelper(arr, low, high);
            //pivot已经找到应该呆的下标,所以不需要包含postion
            quickSort(arr,low,postion-1);
            quickSort(arr,postion+1,high);
        }
        return arr;
    }
	//返回pivot正确的下标,排序之后的下标
    private static int quickSortHelper(int[]arr,int low,int high){
        int pivot=arr[high];
        int slow =low;
        int fast=low;
        while (fast<high){
            if (arr[fast]<=pivot){
                int temp=arr[fast];
                arr[fast]=arr[slow];
                arr[slow]=temp;
                slow++;
            }
            fast++;
        }
        int temp = arr[slow];
        arr[slow] = arr[high];
        arr[high] = temp;
        return slow;
    }