做道简单一点的题巩固一下

基本思路

与上一个整体相似，通过快速排序将长度为n的整数数列变为从小到大的有序数组，在拿出其中第k个数。

题目描述
给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。

输入格式
输入共两行。
第一行包含整数 n和k。
第二行包含 n 个整数（所有整数均在 1∼1e9 范围内），表示整个数列。

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

数据范围
1≤n≤100000
1≤k≤n

输入样例
5 3
3 1 2 4 5

输出样例
3

具体实现

1. 模板
实现思路：
我们假设小于分界点的个数为sl，大于分界点的个数为sr。

此时，整体便会分为两种情况：

当k≤sl时，第k小的数一定在分界点的左边，此时只需要对左边进行递归排序，第k小的数在sl上仍为第k个数。
当k≥sl时，第k小的数一定在分界点的右边，此时只需要对右边进行递归排序，第k小的数在sr上变为第 k-sl 个数。
代码注解：
在C++当中，如果局部变量和全局变量重名时，会优先使用局部变量，因此可以重名。

因为 1≤k≤n ，所以如果 l≥r 的话，就说明整个数列当中只有一个数，此时这个数便是我们的答案，因此返回 q[l]，q[r] 均可。

从 l 到 j 一共有 j-l+1 个数，此时让k与其进行比较，决定我们应该对哪部分区间进行递归处理（具体可见上面实现思路）。

实现代码
 

#include <bits/stdc++.h>
using namespace std;const int N=100010;
int n,k;
int q[N]; 
int quick_sort(int q[],int l,int r,int k)
{if(l>=r){return q[l];}int x=q[l+r>>1],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]);}}if(j-l+1>=k){return quick_sort(q,l,j,k);  }else{return quick_sort(q,j+1,r,k-(j-l+1));}
}
int main()
{cin>>n>>k;for(int i=0;i<n;i++){cin>>q[i];}cout<<quick_sort(q,0,n-1,k)<<endl;system("pause");return 0;
}

2. STL

代码注解：

实现代码

#include <bits/stdc++.h>
using namespace std;const int N=100010;
int n,k;
int q[N]; 
int main()
{cin>>n>>k;for(int i=0;i<n;i++){cin>>q[i];}sort(q,q+n);cout<<q[k-1]<<endl;system("pause");return 0;
}