剑指offer系列源码-最小的K个数

标签：blog   io   os   sp   for   2014   log   ad   ef   

题目1371：最小的K个数
时间限制：1 秒内存限制：32 兆特殊判题：否提交：5175解决：1092
题目描述：
输入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

#include <iostream>
#include<stdio.h>
using namespace std;
//找基准位置
int adjust(int a[],int left,int right){
    int x = a[left];
    while(left<right){
        while(left<right&&a[right]>x){
            right--;
        }
        if(left<right){
            a[left] = a[right];
            left++;
        }
        while(left<right&&a[left]<x){
            left++;
        }
        if(left<right){
            a[right] = a[left];
            right--;
        }
    }
    a[left] = x;
    return left;
}
//递归（使基准位置区间为一个数为止）
void quick_sort(int a[],int left,int right){
    if(left<right){
        int i = adjust(a,left,right);
        quick_sort(a,left,i-1);
        quick_sort(a,i+1,right);
    }
}
int main(){
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        int* a = new int[n];
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        quick_sort(a,0,n-1);
        for(int i=0;i<m;i++){
            if(i==m-1){
               printf("%d\n",a[i]);
            }else{
                printf("%d ",a[i]);
            }
        }
    }
    return 0;
}

剑指offer系列源码-最小的K个数

标签：blog   io   os   sp   for   2014   log   ad   ef   

原文地址：http://blog.csdn.net/hackcoder/article/details/41822667