利用移除元素思想解决快排
在学到快排的时候感觉快排很难,现在捋顺之后,思路廓然开朗了。
先来看一道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
覆盖掉。
但是这跟快排有什么关系呢?
快排思想是:给定一个中心点pivot
将pivot
放到合适的位置,自然就是左边都比pivot
小,右边都比pivot
大。
那么怎么找到左边都比pivot小的下标呢,上面的代码已经说明了一切,只需要定义两个指针:一个是fast,
一个是slow
来保存数组索引
fast
负责遍历,slow
符合记录比pivot
大的元素。当fast找到符合条件的就把fast
和slow
两个索引对应的元素进行交换,因为这是排序。
当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;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix