求第k小的数

来源:互联网 发布:python 私有属性 知乎 编辑:程序博客网 时间:2024/06/02 11:18
题目是这样的,一个无序的数组让你找出第k小的元素,我当时看到这道题的时候也像很多人一样都是按普通的思维,先排序然后取第K个,但是当数组非常大的时候,效率不高,那有没有简单的方法了,其实我们早就学过,只是我们不善于思考和变通。

很多人刚开始非常热衷于各种排序算法只是了解却没深究,这个题目的复杂度是O(n),原理就是快速排序里面的划分算法。


分析:快速排序选择一个base(一般是第一个数)对数组进行划分,左边小于base,右边大于等于base,所以我们计算左边小于base(加上base)的个数总共有多少,如果等于k,正是我们所要的,如果大于k,说明第k小的数在左边,那就在左边进行我们的递归;
否则,在右边,在右边进行我们的递归。

完整代码:

#include<iostream>#include<algorithm>using namespace std;int a[10] = { 2,3,4,1,5,10,9,7,8,6 };int FindKMin(int k){int s = 0;int left, right;int i = 0, j = 9;while (s != k){left = i;right = j;while (i < j){while (i < j&&a[left] <= a[j]) j--;while (i < j&&a[left] >= a[i]) i++;if (i < j)  swap(a[i], a[j]);}swap(a[left], a[i]);//真是好粗心,这里写成了swap(temp,a[i])了,谨记!!!s = i + 1;if (s == k)return a[i];else if (s < k)i++, j = right;elsej--, i = left;}}int main(){int k = 1;printf("第%d小的数是:%d\n", k, FindKMin(k));return 0;}



1 0
原创粉丝点击