分治04--最小的K个数

Posted 2022-08-29 昕光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 := []int4, 5, 1, 6, 2, 7, 3, 8
    k := 4
    ret := GetLeastNumbersV2(arr, k)
    fmt.Println(ret)

注意事项

1. 基于分区的方法待完成…

说明

1. 当前使用 go1.15.8
2. 参考​​牛客网--剑指offer​​ 标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目，例如 jz01 代表牛客网剑指offer中01号题目。

注意!!!

• 笔者最近在学习 golang，因此趁机通过数据结构和算法来进一步熟悉下go语言
• 当前算法主要来源于剑指 offer，后续会进一步补充 LeetCode 上重要算法，以及一些经典算法
• 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解