快速排序 - AcWing 786. 第k个数（C++）- 简单

JIngles123

于 2022-03-08 22:51:34 发布

阅读量300

点赞数

分类专栏： # 算法基础课-1、基础算法文章标签： c++ 排序算法算法

本文链接：https://blog.csdn.net/qq_40467670/article/details/123365387

版权

算法基础课-1、基础算法专栏收录该内容

11 篇文章 1 订阅

订阅专栏

该博客主要介绍了C++实现的快速选择算法，用于在未排序的整数数组中找到第k小（或大）的元素。代码示例展示了如何在主函数中调用并使用该算法，并通过递归进行部分排序。此算法在处理大数据集时效率较高，适用于中等规模的数据处理任务。

摘要由CSDN通过智能技术生成

题目链接：https://www.acwing.com/problem/content/description/788/
题目如下：
在这里插入图片描述

#include <iostream>
#include <algorithm>
using namespace std;
const int N=100010;
int arr[N];
int main(){
    int n,k;
    cin>>n>>k;
    for(int i=0;i<n;i++) cin>>arr[i];
    cout<<quick_select(arr,0,n-1,k)<<endl;
    return 0;
}
void quick_select(int arr[],int l,int r,int k){
    if(l==r) return arr[l];
    int partval=arr[(l+r)>>1];
    int i=l-1,j=r+1;
    while(i<j){
        do ++i;while(arr[i]<partval);
        do --j;while(arr[j]>partval);
        if(i<j) swap(arr[i],arr[j]);
    }
    int len=j-l+1;
    if(k<=len) return quick_select(arr,l,j,k);
    return quick_select(arr,j+1,r,k-len);
}