分治04--最小的K个数

题目概述

• 算法说明
  给定一个数组，找出其中最小的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 GetLeastNumbersSolution(input []int, k int) []int {
	// write code here
	if k <= 0 || k > len(input) {
		return []int{}
	}
	sort.Ints(input)
	return input[:k]
}

func Partition(input []int, p, r int) int {
	pivot := input[r]
	i := p
	for j := p; j < r; j++ {
		if input[j] < pivot {
			input[i], input[j] = input[j], input[i]
			i++
		}
	}
	input[i], input[r] = input[r], input[i]
	return i
}

func GetLeastNumbersSolutionV2(input []int, k int) []int {
	if k <= 0 || k > len(input) {
		return []int{}
	}
	start, end := 0, len(input)-1
	index := Partition(input, start, end)
	for index != k-1 {
		if index > k-1 {
			end = index - 1
			index = Partition(input, start, end)
		} else {
			start = index + 1
			index = Partition(input, start, end)
		}
	}
	return input[:k]
}

type DataCase struct {
	Arr []int
	K   int
}

func TestGetLeastNumbersSolution() {
	arrs := []DataCase{
		{[]int{4, 5, 1, 6, 2, 7, 3, 8}, 4},
		{[]int{1}, 0},
		{[]int{0, 1, 2, 1, 2}, 3},
	}
	for _, arr := range arrs {
		fmt.Printf("arr=%v,k=%v,result=%v\n", arr.Arr, arr.K, GetLeastNumbersSolution(arr.Arr, arr.K))
	}

	fmt.Println("version2 with partition:")
	for _, arr := range arrs {
		fmt.Printf("arr=%v,k=%v,result=%v\n", arr.Arr, arr.K, GetLeastNumbersSolutionV2(arr.Arr, arr.K))
	}
}

func main() {
	TestGetLeastNumbersSolution()
}

注意事项

1. golang中 sort.Ints 默认使用 Pattern-defeating Quicksort, pdqsort

说明

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

注意!!!

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