快速排序及TopK相关题目解析

最新推荐文章于 2024-07-01 10:58:21 发布

原创最新推荐文章于 2024-07-01 10:58:21 发布 · 315 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++ #后端

刷题专栏收录该内容

20 篇文章

订阅专栏

本文介绍了使用快速排序算法解决LeetCode上的四个问题，包括颜色划分、数组排序、找出第K个最大元素以及最小K个数，通过代码示例详细解析了原理和实现过程。

文章目录

简介
颜色划分
- 原理
- 代码
排序数组(重要)
- 原理
- 代码
数组中的第K个最大元素
- 原理
- 代码
最小K个数
- 原理
- 代码

简介

大家好,这里是jiantaoyab,这篇文章给大家带来的是快排类型相关的题目练习和解析,希望大家能相互讨论进步

颜色划分

https://leetcode.cn/problems/sort-colors

原理

在这里插入图片描述

代码

class Solution {
public:
    void sortColors(vector<int>& nums) {
       int n = nums.size();
       int left = -1;
       int right = n ;
       for(int i = 0; i < right; )
       {
           if(nums[i] == 0)
           {
               swap(nums[++left],nums[i++]);
           }
           else if(nums[i] == 1)
           {
               i++;
           }
           else
           {
               swap(nums[--right],nums[i]);
           }
       }
    }
};

排序数组(重要)

https://leetcode.cn/problems/sort-an-array/

原理

在这里插入图片描述

代码

class Solution {
public:
    int GetRandom(vector<int>& nums, int left, int right)
    {
        return nums[rand() % (right - left + 1) + left];
    }
    void Qsort(vector<int>& nums , int l, int r)
    {
        if(l >= r) return;
        int key = GetRandom(nums, l, r);
        unsigned int i = l,left = l - 1, right = r + 1 ;
        while(i < right)
        {           
            if(nums[i] < key) 
                swap(nums[++left],nums[i++]);
            else if(nums[i] == key) 
                i++;
            else 
                swap(nums[--right],nums[i]);
        }
        //[l, left],[left+1 ,right-1] ,[right , r]
        Qsort(nums,l,left);
        Qsort(nums,right,r);
    }
    vector<int> sortArray(vector<int>& nums) {
        srand((unsigned)time(NULL)); //种下随机数种子
        Qsort(nums, 0, (int)nums.size() - 1);
        return nums;
    }
};

数组中的第K个最大元素

https://leetcode.cn/problems/kth-largest-element-in-an-array/

原理

在这里插入图片描述

代码

class Solution {
public:
    int GetRandom(vector<int>&nums, int left, int right)
    {
        return nums[rand() % (right - left + 1) + left];
    }
    int Qsort(vector<int>&nums, int l, int r, int k)
    {
        if( l == r) return nums[l]; //区间只有一个元素返回
        //1.随机选择基准
        int key = GetRandom(nums, l, r);
        int i = l, left = l - 1, right = r + 1;
        //2.划分区间
        while(i < right)
        {
            if(nums[i] < key) 
                swap(nums[++left],nums[i++]);
            else if(nums[i] == key)
                i++;
            else 
                swap(nums[--right],nums[i]);
        }
        //[l, left] [left + 1, right - 1] ,[right,r]
        int z = r - right + 1;
        int y = right - left - 1;
        if(z >= k) 
        return Qsort(nums,right,r,k);
        else if(z + y >= k) 
            return key;
        else
        return Qsort(nums,l,left,k-y-z);
    }
    int findKthLargest(vector<int>& nums, int k) {
        srand((unsigned)time(nullptr));//随机数种子
        return Qsort(nums,0,nums.size()-1,k);
    }
};

最小K个数

https://leetcode.cn/problems/smallest-k-lcci/

原理

在这里插入图片描述

代码

class Solution {
public:
    int GetRondam(vector<int>& arr, int left, int right)
    {
        return arr[rand() % (right - left + 1) + left];
    }
    void  Qsort(vector<int>& arr, int l, int r, int k)
   {
       if(l >= r)  return ;
       int key = GetRondam(arr, l, r);
       int i = l, left = l - 1, right = r + 1;
       while( i < right)
       {
           if(arr[i] < key) 
                swap(arr[++left],arr[i++]);
           else if(arr[i] == key)
                i++;
           else
                swap(arr[--right],arr[i]);
       }
        //[l, left] [left + 1, right - 1] ,[right,r]
       int x =  left - l + 1;
       int y = right - left - 1;
       if(x > k) return Qsort(arr, l, left, k);
       else if( x + y >= k) return ;
       else return Qsort(arr, right, r, k - x - y);

   }
    vector<int> smallestK(vector<int>& arr, int k) {
        srand((unsigned)time(nullptr));
        Qsort(arr, 0, arr.size()-1, k);
        return {arr.begin(),arr.begin() + k};
    }
};