QuickSelect 查找第 K 小的元素

最新推荐文章于 2024-08-26 09:08:17 发布

shida_csdn

最新推荐文章于 2024-08-26 09:08:17 发布

阅读量783

点赞数

分类专栏： Golang 经典题目编程之美 - Golang 文章标签： QuickSelect 查找第 K 小的元素

本文链接：https://blog.youkuaiyun.com/shida_csdn/article/details/85091219

版权

Golang 同时被 3 个专栏收录

51 篇文章

订阅专栏

经典题目

47 篇文章

订阅专栏

编程之美 - Golang

9 篇文章

订阅专栏

本文深入解析了Quickselect算法，一种用于未排序数组中查找第k小元素的有效方法。该算法基于快速排序，通过多次分区操作直到找到目标元素，提供了一个高效且实用的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Quick select 算法通常用来在未排序的数组中寻找第 k 小/第 k 大的元素。其方法类似于 Quick sort。

本质上是通过多次快速排序，当某次快速排序的枢纽元素恰好下标为 k-1 时，结束查找~

package main

import "fmt"

func core(nums []int, k, start, end int) int {
	left, right := start, end
	key := nums[left]

	for left < right {
		for left < right && nums[right] >= key {
			right --
		}
		nums[left] = nums[right]
		for left < right && nums[left] <= key {
			left ++
		}
		nums[right] = nums[left]
	}
	nums[left] = key //left 是下标，k 也是下标

	if left < k {
		return core(nums, k, left+1, end)
	} else if left > k {
		return core(nums, k, start, left-1)
	}
	return nums[left]
}

func findKSmallestNum(nums []int, k, start, end int) int {
	return core(nums, k-1, start, end)
}
func main() {
	nums := []int{1, 3, 2, 6, 5, 4}
	result := findKSmallestNum(nums, 3, 0, 5)
	fmt.Println(result)
}

注意快速排序的写法，注意 k 和 left 均为下标