golang双指针快速排序的实现代码

package main
import "fmt"
func QuickSort(arr []int) {
	if len(arr) <= 1 {
		return
	}
	// 选中间位置作为基准，比它值小的移动到左边，比它大的值移动到右边
	pivotIndex := len(arr) / 2
	pivot := arr[pivotIndex]
	arr[pivotIndex], arr[len(arr)-1] = arr[len(arr)-1], arr[pivotIndex]
	left := 0
	right := len(arr) - 2
	for left <= right { // 每一轮移动一次指针
		if arr[left] <= pivot {
			left++
		} else {
			arr[left], arr[right] = arr[right], arr[left]
			right--
		}
	}
	pivotIndex = left
	arr[left], arr[len(arr)-1] = arr[len(arr)-1], arr[left]
	QuickSort(arr[:pivotIndex])
	QuickSort(arr[pivotIndex+1:])
}
func main() {
	arr := []int{4, 1, 3, 5, 2}
	QuickSort(arr)
	fmt.Println(arr)
}

func quickSort(data []int) {
//如果数组长度为1，返回
	if len(data) <= 1 {
		return
	}
	//设置基准为数组第一个元素
	base := data[0]
	//两个指针分别指向待交换首尾位置
	l, r := 0, len(data)-1
	//基准为第一个元素，比较的元素从第二个开始
	for i := 1; i <= r; {
	//如果比较元素大于基准元素，把该元素放到数组尾部，把尾部元素交换过来
	//此时尾部的元素已经判断过比基准元素大，因此r--向前移动，
	//而i所在的元素为新交换过来的还没有判断过大小，因此保持不动
		if data[i] > base {
			data[i], data[r] = data[r], data[i]
			r--
		} else {
		//如果比较元素小于等于基准元素，则交换当前元素和基准元素的位置
		//l指向的是基准元素，因此l++才能重新指向基准元素,i++判断下一个数
			data[i], data[l] = data[l], data[i]
			l++
			i++
		}
	}
	//此时l指向基准元素，l前面的元素都比l小，l后面的都比l大
	//因此对l前面和后面的数组再次快排，直到子数组长度为1结束
	quickSort(data[:l])
	quickSort(data[l+1:])
}

func mergeSort(data []int) []int {
	length := len(data)
	//如果长度为1不可再分，就返回
	if length <= 1 {
		return data
	}
	//把数组一分为二
	num := length / 2
	//左边和右边的数组都要再分
	left := mergeSort(data[:num])
	right := mergeSort(data[num:])
	//对于每对数组进行排序
	return merge(left, right)
}
func merge(left, right []int) (result []int) {
//数组是从长度为1开始进行添加的，因此每个子数组都是有序的
	l, r := 0, 0
	//当两个数组都没有遍历完的时候，按照元素大小依次添加
	for l < len(left) && r < len(right) {
		if left[l] < right[r] {
			result = append(result, left[l])
			l++
		} else {
			result = append(result, right[r])
			r++
		}
	}
	//一个数组遍历完比，把没有遍历完的数组直接添加进去
	result = append(result, left[l:]...)
	result = append(result, right[r:]...)
	return
}

func heapSort(array []int) {
	m := len(array)
	s := m / 2
	for i := s; i > -1; i-- {
		heap(array, i, m-1)
	}
	for i := m - 1; i > 0; i-- {
		array[i], array[0] = array[0], array[i]
		heap(array, 0, i-1)
	}
}
func heap(array []int, i, end int) {
	l := 2*i + 1
	if l > end {
		return
	}
	n := l
	r := 2*i + 2
	if r <= end && array[r] > array[l] {
		n = r
	}
	if array[i] > array[n] {
		return
	}
	array[n], array[i] = array[i], array[n]
	heap(array, n, end)
}