AcWing习题课（一）（786、788、790、795、796、797、798）

谢耳朵(wer~wer~)

于 2025-06-06 15:19:46 发布

阅读量630

点赞数 27

分类专栏： Acwing学习文章标签：算法数据结构

本文链接：https://blog.csdn.net/a3040218/article/details/148459584

版权

本章的具体知识点可以看我前面的算法基础课2

786--第k个数

给定一个长度为n的整数数列，以及一个整数k，请用快速选择算法求出数列的第k小的数是多少。

输入格式

第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在1~109109范围内），表示整数数列。

输出格式

输出一个整数，表示数列的第k小数。

数据范围

1≤n≤1000001≤n≤100000,
1≤k≤n1≤k≤n

输入样例：

5 3
2 4 1 5 3
输出样例：

3

使用快速排序，然后直接输出从小到大排序后的第 k 个数。时间复杂度为O(nlogn)

①分界点x

②左边都<=x,右边都>=x

③分别递归左右两边

..具体看我前面的文章..

左边个数有SL，右边SR

若K<=SL，第K小数字一定在左边，只要递归左边

若K>=SR，递归右边（k-SL）

#include <iostream>
using namespace std;

const int N=100010;
int n;
int k;
int q[N];

void quick_sort(int q[],int l,int r){
    if(l>=r) return;
    int x=q[(l+r)/2],i=l-1,j=r+1;
    while(i<j){
        do i++; while(q[i]<x);
        do j--; while(q[j]>x);
        if(i<j) swap(q[i],q[j]);
    }
    quick_sort(q,l,j);
    quick_sort(q,j+1,r);
    return;
}

int main(){
    scanf("%d %d",&n,&k);
    for(int i=0;i<n;i++){
        scanf("%d",&q[i]);
    }
    quick_sort(q,0,n-1);
    printf("%d",q[k-1]);
    return 0;
}