快速选择算法用于查找元素,当列表进行排序时,该元素将位于 Kth 位置,而无需实际排序整个列表。快速选择与快速排序非常相似,可以代替对整个列表进行排序,我们只是忽略快速排序每一步列表的一半,只关注我们感兴趣的列表区域。
package main
import (
"fmt"
)
func QuickSelect(arr []int, key int) int {
size := len(arr)
QuickSelectUtil(arr, 0, size-1, key)
return arr[key-1]
}
func QuickSelectUtil(arr []int, lower int, upper int, key int) {
if upper <= lower {
return
}
pivot := arr[lower]
start := lower
stop := upper
for lower < upper {
for arr[lower] <= pivot && lower < upper {
lower++
}
for arr[upper] > pivot && lower <= upper {
upper--
}
if lower < upper {
swap(arr, upper, lower)
}
}
swap(arr, upper, start) // upper is the pivot position
QuickSelectUtil(arr, start, upper-1, key) // pivot -1 is the upper for left sub array.
QuickSelectUtil(arr, upper+1, stop, key) // pivot + 1 is the lower for right sub array.
}
func swap(arr []int, first int, second int) {
arr[first], arr[second] = arr[second], arr[first]
}
func main() {
data := []int{9, 1, 8, 2, 7, 3, 6, 4, 5}
fmt.Println(QuickSelect(data, 5))
}