[九度][何海涛] 最小的K个数

题目描述：

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

输入：

每个测试案例包括2行：

第一行为2个整数n，k(1<=n，k<=200000)，表示数组的长度。

第二行包含n个整数，表示这n个数，数组中的数的范围是[0,1000 000 000]。

输出：

对应每个测试案例，输出最小的k个数，并按从小到大顺序打印。

样例输入：

8 4
4 5 1 6 2 7 3 8

样例输出：

1 2 3 4

可以用一个K大小的最大队来做，当堆的个数<K时放入，否则当堆的根大于当前数，弹出堆的根，插入当前数。另外一种就是用partition来找第K大数，这样它前面的保证都是比它小的，所以K个数就找到了。

#include <iostream>
#include <vector>
#include <cstdio>
#include <queue>
using namespace std;

void print(priority_queue<int, vector<int>, less<int> > &maxHeap)
{
    if (maxHeap.empty())
        return;

    int element = maxHeap.top();
    maxHeap.pop();
    bool status = maxHeap.empty();

    print(maxHeap);

    if (status)
        printf("%d", element);
    else
        printf(" %d", element);
}

int main()
{
    int n, k;

    while(scanf("%d%d", &n, &k) != EOF)
    {
        priority_queue<int, vector<int>, less<int> > maxHeap;

        int num;
        for(int i = 0; i < n; i++)
        {
            scanf("%d", &num);
            if (maxHeap.size() < k)
                maxHeap.push(num);
            else
            {
                if (maxHeap.top() > num)
                {
                    maxHeap.pop();
                    maxHeap.push(num);
                }
            }
        }

        print(maxHeap);
        printf("\n");
    }
}

#include <iostream>
#include <ctime>
#include <algorithm>
#include <cstdio>
using namespace std;

void partition(int a[], int left, int right, int k)
{
    int randNum = left + (rand() % (right - left + 1));
    int t = a[randNum];
    a[randNum] = a[left];
    a[left] = t;

    int i = left;
    int j = right;
    int key = a[left];
    while(i <= j)
    {
        if (a[i] <= key)
            i++;
        else
        {
            int t = a[j];
            a[j] = a[i];
            a[i] = t;

            j--;
        }
    }

    a[left] = a[j];
    a[j] = key;

    if (j == k)
        return;
    else if (j > k)
        return partition(a, left, j - 1, k);
    else
        return partition(a, j + 1, right, k);
}

int main()
{
    int n, k;
    int a[200000];

    while(scanf("%d%d", &n, &k) != EOF)
    {
        for(int i = 0; i < n; i++)
            scanf("%d", &a[i]);

        srand(time(NULL));
        partition(a, 0, n - 1, k - 1);

        sort(a, a + k);

        for(int i = 0; i < k; i++)
            if (i == 0)
                printf("%d", a[i]);
            else
                printf(" %d", a[i]);
        printf("\n");
    }
}