《剑指offer》面试题30:
题目:输入n个整数,找出其中最小的k个数。例如输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
分析:求数组的第k小,如果能改变数组,那么采用二分法;
如果求数组前k小,如果可以改变数组,同样可以采用二分法;
如果不能改变数组那么采用堆排序,大堆的查找最大数操作复杂度为O(1),插入和删除操作复杂度为O(logk),也可以采用红黑树,红黑树是一种很有用的数据结构,在STL中广泛应用,红黑树是一种平衡二叉树,它的查找、插入和删除操作复杂度都为O(logk),如果可以直接采用STL中的set和multiset,set不允许键值重复,而multiset允许键值重复。
关于更多k最小的问题看以前一篇日志:两个排序数组合并第k或前k个最小值问题
(1)二分法,O(n)复杂度,会改变数组
void GetLeastNumbers_Solution1(int* input, int n, int* output, int k)
{
if(input == NULL || output == NULL || k > n || n <= 0 || k <= 0)
return;
int start = 0;
int end = n - 1;
int index = Partition(input, n, start, end);
while(index != k - 1)
{
if(index > k - 1)
{
end = index - 1;
index = Partition(input, n, start, end);
}
else
{
start = index + 1;
index = Partition(input, n, start, end);
}
}
for(int i = 0; i < k; ++i)
output[i] = input[i];
}
(2)红黑树法,复杂度O(nlogk),不改变数组
typedef multiset<int, greater<int> > intSet;
typedef multiset<int, greater<int> >::iterator setIterator;
void GetLeastNumbers_Solution2(const vector<int>& data, intSet& leastNumbers, int k)
{
leastNumbers.clear();
if(k < 1 || data.size() < k)
return;
vector<int>::const_iterator iter = data.begin();
for(; iter != data.end(); ++ iter)
{
if((leastNumbers.size()) < k)
leastNumbers.insert(*iter);
else
{
setIterator iterGreatest = leastNumbers.begin();
if(*iter < *(leastNumbers.begin()))
{
leastNumbers.erase(iterGreatest);
leastNumbers.insert(*iter);
}
}
}
}