最小的K个数

题目描述
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。
思路：基本排序

# -*- coding:utf-8 -*-
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        if not tinput or k<=0 or k>len(tinput):
            return []
        tinput.sort()
        res=[]
        for i in range(k):
            res.append(tinput[i])
        return res

时间复杂度O(nlogn)

import java.util.ArrayList;
import java.util.Arrays;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list=new ArrayList<>();
        if (input==null||k<=0||k>input.length) return list;
        Arrays.sort(input);
        for(int i=0;i<k;i++) list.add(input[i]);
        return list;
    }
}

python大顶堆模块nsmallest

# -*- coding:utf-8 -*-
import heapq
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        if not tinput or k<=0 or k>len(tinput):
            return []
        return heapq.nsmallest(k, tinput, key=None)

排序可以用快排

堆排序 时间复杂度O(nlogk)

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> result= new ArrayList<>();
        if(input.length==0 || k>input.length || k<=0)
            return result;
        int[] numbers=new int[k];//存最小的k个数
        for(int i=0;i<k;i++) numbers[i]=input[i];
        //将前k个数组构造成最大堆形式
        for(int i=k/2-1;i>=0;i--){
            heapadjust(numbers,i,k-1);
        }
        //将剩余的进行比较调整，存在更小的则重新调整
        for(int i=k;i<input.length;i++){
            if (input[i]<numbers[0]){
                numbers[0]=input[i];
                heapadjust(numbers,0,k-1);//重新调整最大堆
            }
        }
        for (int n:numbers)
            result.add(n);
        return result;
    }
    private void heapadjust(int[] arr,int start,int end){
        int temp=arr[start];//根值
        int child=start*2+1;//孩子节点
        while (child<=end){//找到孩子节点中的最大值
            if(child+1<=end && arr[child]<arr[child+1])
                child++;
            if(temp<arr[child]){
                arr[start]=arr[child];//交换后进行下一次循环
                start=child;//修改根节点为当前节点
                child=start*2+1;//修改孩子节点
            }
            else break;
        }
        arr[start]=temp;
    } 
}