菜单 学习猿地 - LMONKEY

VIP

开通学习猿地VIP

尊享10项VIP特权 持续新增

知识通关挑战

打卡带练!告别无效练习

接私单赚外块

VIP优先接,累计金额超百万

学习猿地私房课免费学

大厂实战课仅对VIP开放

你的一对一导师

每月可免费咨询大牛30次

领取更多软件工程师实用特权

入驻
194
0

特定分治查找第K小 BFPRT算法

原创
05/13 14:22
阅读数 88164

1、问题

使用特定的分治策略去寻找无序数组中第 k 小的元素。

2、解析

step1:数组被划分为了 N/5 个小部分

step2:取出每个小部分的中位数,一共有 N/5 个,找出这些数的中位数,记为 pivot

step3:以 pivot 作为比较,将整个数组划分为 <pivot , =pivot , >pivot 三个区域

step4: 判断第K小的数在哪个区域,如果在 = 区域则直接返回 pivot ,如果在 < 或 > 区域,则将这个区域的数递归调用

3、设计

 1 /* 区间快排,各组返回中位数下标 */
 2 int GetPivotIndex(int left, int right) {
 3     sort(Array + left, Array + right + 1)
 4     return (right - left) >> 1) + left
 5 }
 6 
 7 /* 返回中位数的中位数下标 */
 8 int GetPivotOfPivotIndex(int left, int right) {
 9     if right - left < 5
10         return GetPivotIndex(left, right)
11     pos <- left - 1
12     for i <- left  and i + 4 <= right  and i += 5
13         index <- GetPivotIndex(i, i + 4)  //找到五个元素的中位数的下标
14         swap(Array[++pos], Array[index])   //依次放在左侧
15         
16     return KthNumber(left, pos, ((pos - left + 1) >> 1) + 1)
17 }
18 
19 /* 利用中位数的中位数的下标进行划分,返回分界线下标 */
20 int Partition(int left, int right, int pivot_index) {
21     swap(Array[pivot_index], Array[right])  //把基准放置于末尾
22     divide_index <- left  //跟踪划分的分界线
23     for i <- left to right
24         if Array[i] < Array[right]
25             swap(Array[divide_index++], Array[i]) //比基准小的都放在左侧
26     }
27     swap(Array[divide_index], Array[right]);  //最后把基准换回来
28     return divide_index
29 }
30 
31 int KthNumber(int left, int right, const int& k) {
32     pivot_index <- GetPivotOfPivotIndex(left, right)   //得到中位数的中位数下标
33     divide_index <- Partition(left, right, pivot_index)  //进行划分,返回划分边界
34     num <- divide_index - left + 1
35     if (num == k)//如果分界线刚好在k位置,直接返回该下标
36         return divide_index
37     else if (num > k)//如果分界线在k左边,递归左区间
38         return KthNumber(left, divide_index - 1, k)
39     else //如果分界线在k右边,递归右区间,并修改k值
40         return KthNumber(divide_index + 1, right, k - num)
41 }

4、分析

5、源码

https://github.com/ChenyuWu0705/Algorithm-Analyze-and-Design/blob/main/TOP-K.cpp

发表评论

0/200
194 点赞
0 评论
收藏
为你推荐 换一批