题目是这样的,一个无序的数组让你找出第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;
else
j--, i = left;
}
}
int main()
{
int k = 1;
printf("第%d小的数是:%d\n", k, FindKMin(k));
return 0;
}