1.移动零(双指针)
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:
输入: nums = [0]
输出: [0]
采用数组分块的思想,将数组分为三块
[0,left] 非零元素;
[left+1,i] 零元素;
[i,n-1] 待处理元素;
实现代码如下;
class Solution {
public:
void moveZeroes(vector<int>& nums) {
for(int cur=0,dest=-1;cur<nums.size();cur++)
{
if(nums[cur]!=0)
{
dest++;
swap(nums[dest],nums[cur]);
}
}
}
};
2.快速排序三路划分
class Solution {
public int qsort(int[]nums,int l,int r,int k)
{
if(l==r)
return nums[l];
int key=nums[new Random().nextInt(r-l+1+l)];
int left=l-1,right=r+1,i=l;
//[l,left] [left+1,right-1] [right,r]
while(i<right)
{
if(nums[i]<key) swap(nums,++left,i++);
else if(nums[i]==key) i++;
else swap(nums,--right,i);
}
qsort(nums,l,left);
qsort(nums,right,r);
}
public void swap(int[]nums,int x,int y)
{
int tmp=nums[x];
nums[x]=nums[y];
nums[y]=tmp;
}
}
采用三个指针left,right,i将数组划分为三个部分
[l,left] 小于key的元素;
[left+1,right-1] 等于key的元素;
[right,r] 大于key的元素;
然后分别对左右区间进行递归排序;
3.颜色划分
快速排序三路划分的变式,核心是数组划分;
题目描述:
给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/sort-colors
对数组分三块,小于1的放左边,等于1的在中间,大于1的放右边;
[l,left] 小于1的元素;
[left+1,right-1] 等于1的元素;
[right,r] 大于1的元素;
4.快速选择算法解决TOPK问题
题目描述:
给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/kth-largest-element-in-an-array
本题有多种不同的思路:
1.排序,返回nums[n-k]; 时间复杂度N*logN;
2.堆排序:
3.快速选择算法;
下面我们重点说一下快速选择算法;
class Solution {
public int findKthLargest(int[] nums, int k) {
return qsort(nums,0,nums.length-1,k);
}
public int qsort(int[]nums,int l,int r,int k)
{
if(l==r)
return nums[l];
int key=nums[new Random().nextInt(r-l+1+l)];
int left=l-1,right=r+1,i=l;
//[l,left] [left+1,right-1] [right,r]
while(i<right)
{
if(nums[i]<key) swap(nums,++left,i++);
else if(nums[i]==key) i++;
else swap(nums,--right,i);
}
int b=right-left-1,c=r-right+1;
if(c>=k)
return qsort(nums,right,r,k);
else if(b+c>=k)
return key;
else return qsort(nums,l,left,k-b-c);
}
public void swap(int[]nums,int x,int y)
{
int tmp=nums[x];
nums[x]=nums[y];
nums[y]=tmp;
}
}
核心在于分类讨论,在快速排序三路划分的基础上
[l,left] 小于key的元素; a
[left+1,right-1] 等于key的元素;b
[right,r] 大于key的元素;c
if(c>=k) 继续在右区间寻找第K大元素;
if(b+c>k) return key;
else 在左区间寻找第k-b-c大元素;
快速排序是一个非常非常重要的排序算法,核心思想就是数组分块加分治,一定要好好掌握!