Go实现快速选择算法

快速选择算法用于查找元素，当列表进行排序时，该元素将位于 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))

}