数组——按值划分区域

metabit

已于 2023-02-19 16:32:57 修改

阅读量357

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构与算法文章标签： go

于 2021-10-23 17:38:40 首次发布

本文链接：https://blog.csdn.net/dawnto/article/details/120923758

数据结构与算法专栏收录该内容

185 篇文章

订阅专栏

该博客探讨了快速排序算法中的Partition过程，其目标是将数组arr根据给定的整数num进行分区，使得小于等于num的元素位于左侧，大于num的元素位于右侧。博主提供了使用Go语言实现的代码示例，确保在O(1)额外空间复杂度和O(N)时间复杂度内完成操作。测试用例包括不同类型的数组输入，展示了算法的灵活性和效率。

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

快速排序 前奏
Partition 过程
给定一个数组arr，和一个整数num。请把小于等于num的数放在数组的左边，大于num的数放在数组的右边。不需要有序
要求：额外空间复杂度O(1),时间复杂度O(N)
         \
  <= 区域 |         [ 5   3   7   2   3    4   1 ]     num = 3
         /    下标    0   1   2   3   4    5   6

1. arr[i] <= num  当前数和 <=区下一个数交换, <= 区右扩,i++
2. arr[i] >  num  i++ 直接跳下一个

func TestP(t *testing.T)  {
	arr := []int{12, 321, 2, 901, 0, 23, 344, 2, 1, -100, -200, -32, 4, 3, 5, 6, 9}
	//arr = []int{1,3,4,5,2}
	arr = []int{ 5,   3,   7,   2,   3,    4,   1}
	fmt.Println(p(arr,3 ))
	fmt.Println(arr)
}


func p(arr []int, num int) int {
	leIndex := -1
	for i := 0; i < len(arr); i++ {
		if arr[i] <= num  {
			leIndex++
			if leIndex == i {
				continue
			}
			arr[i] ^= arr[leIndex]
			arr[leIndex] ^= arr[i]
			arr[i] ^= arr[leIndex]
		}
	}
	return leIndex
}