首页 > 代码库 > 剑指offer系列源码-最小的K个数

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

题目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个数