面试题整理10 最小的k个数

《剑指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);
            }
        }
    }
}