分治04--最小的K个数

原创

昕光xg 2022-08-29 10:03:46 博主文章分类：数据结构与算法 ©著作权

文章标签 最小的K个数剑指offer 数组牛客网 文章分类 运维

©著作权归作者所有：来自51CTO博客作者昕光xg的原创作品，请联系作者获取转载授权，否则将追究法律责任

分治04--最小的K个数-jz29

题目概述
解析&参考答案
注意事项
说明

题目概述

算法说明
给定一个数组，找出其中最小的K个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。如果K>数组的长度，那么返回一个空的数组。
测试用例
输入： [4,5,1,6,2,7,3,8],4
输出： [1,2,3,4]

解析&参考答案

解析
方法1：先sort，然后再获取前K个即可。
方法2：用分区的方式来解决，若基于第K个数字调整，使比第 K 个数字小的都在数组的左边，比第 K 个数字大的所有数字都位于数组的右边。
参考答案

vim jz26.go
package main

import (
    "fmt"
    "sort"
)

func GetLeastNumbersV2(input []int, k int) []int {
    if len(input) == 0 || k > len(input) {
        return []int{}
    }
    sort.Ints(input)
    return input[:k]
}

func GetLeastNumbers(input []int, k int) []int {
    // to add
    return []int{}
}

func main() {
    arr := []int{4, 5, 1, 6, 2, 7, 3, 8}
    k := 4
    ret := GetLeastNumbersV2(arr, k)
    fmt.Println(ret)
}