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

来自:网络
时间:2024-06-07
阅读:

golang双指针快速排序

快速排序算法思想:选中间位置作为基准,比它值小的移动到左边,比它大的值移动到右边。然后把基准值和末位元素交换(方便比较和移动)。定义两个指针left(起始位置0), right(起始位置len(arr) - 2), 循环和基准值比较,每次比较确定一个位置,移动指针,直到不满足left <= right.

注意:快速排序是一种原地排序,不依赖额外空间复杂度,但不是稳定的排序

时间复杂度:平均为O(nlogn), 最慢为O(n^2)。粗略计算时间复杂度主要分为2个部分,分区(O(n))和递归(O(logn)), 相乘为O(nlogn)。
空间复杂度:O(1)

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)
}

扩展:

Golang实现快速排序和归并排序以及堆排序算法全注释

douyin LSY_HELLOWORLD,已成功入职互联网大厂,请关注我,了解非科班的程序员的工作生活把

快速排序算法
快速排序算法步骤如下:

  • 首先从一个数组里找一个基准
  • 对数组进行遍历
  • 把比基准小的放基准左边,把比基准大的放基准右边
  • 以基准为分割点,分成两个数组重复1-4步骤
  • 直到数组长度为1返回
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:])
}

归并排序算法
步骤如下:

  • 把数组二分为left和right
  • 对left和right再次二分
  • 直到数组长度为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)
}
返回顶部
顶部