九度oj 题目1371：最小的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
一开始超时，后来一直在整快排的partition，一直弄不对，参考了以前写的博客，才做出来，汗！

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>

#define MAX 200002

int toDeal[MAX];

int cmp(const void *a, const void *b) {
    int at = *(int *)a;
    int bt = *(int *)b;
    return at-bt;
}


int partition(int start, int end) {
    int base = toDeal[start];
    int sptr = start,eptr = end;
    while(sptr < eptr) {
        while(sptr < eptr && toDeal[eptr] >= base) {
            eptr--;
        }
        toDeal[sptr] = toDeal[eptr];
        while(sptr < eptr && toDeal[sptr] <= base) {
            sptr++;
        }
        toDeal[eptr] = toDeal[sptr];
        
    }
    toDeal[sptr] = base;
    return sptr;
}

void calMinK(int start, int end, int k) {
    int bw = partition(start,end);
    
    if(bw == k - 1) {
        qsort(&toDeal[start], k-start, sizeof(int), cmp);
    }
    else if(bw > k - 1) {
        calMinK(start, bw, k);
    }
    else if(bw < k - 1) {
        qsort(&toDeal[start], bw+1-start, sizeof(int), cmp);
        //2 5 6 8 5 8
        calMinK(bw+1, end, k);
    }
}

int main(int argc, char const *argv[])
{
    int n,k;
    freopen("input.txt","r",stdin);
    while(scanf("%d %d",&n,&k) != EOF) {
        for(int i = 0; i < n; i++) {
            scanf("%d",&toDeal[i]);
        }
        calMinK(0,n-1,k);
        bool isBegin = true;
        for(int i = 0; i < k; i++) {
            if(isBegin) {
                printf("%d",toDeal[i]);
                isBegin = false;
            }
            else {
                printf(" %d",toDeal[i]);
            }
            
        }
        puts("");
    }
    return 0;
}