算法分类:
分治
算法原理:
每次丢弃一部分数组中元素,则问题规模以几何级数递减。
算法步骤:
1.对数组进行分组,分成小于data[k]元素,data[k],以及大于data[k]元素3组。
2.判断小于data[k]的元素个数num。
(1) 若num+1 == k,则返回data[k](获得答案)
(2) 若num + 1 < k,则对大于data[k]组元素递归,同时更改相应k值。
(3) 若num + 1 > k,则对小于data[k]组递归。
时空复杂度:
时间复杂度O(n)
代码实现:
#include <iostream>
using namespace std;
int FindMinK(int data[], int l, int r, int k)
{
int mm, i, j, temp, num;
// 获取目前第k小元素
mm = data[l+k-1];
// 将该元素放到数组末尾
swap(data[r],data[l+k-1]);
i = j = l;
while (j < r)
{
if(data[j] < mm)
{
swap(data[i],data[j]);
++ i;
}
++ j;
}
swap(data[i],data[j]);
num = i-l+1;
if (k == num)
return mm;
else if (k > num)
return FindMinK(data, i+1, r, k-num);
else
return FindMinK(data, l, i-1, k);
};
int main()
{
//for (int i = 1; i <= sizeof(data)/sizeof(data[0]); ++i)
// printf("%d\n", findKMin(data, 0, 8, i));
for (int i = 1; i <= 8; ++ i)
{
int data[]={8,6,6,4,4,3,2,1};
printf("%d\n", FindMinK(data, 0, 7, i));
}
getchar();
return 0;
}