快速排序算法思想及相关常见算法题

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大元素；
 

快速排序是一个非常非常重要的排序算法，核心思想就是数组分块加分治，一定要好好掌握！