BFPRT算法Python实现

    <span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);">BFPRT,通常叫做Median of Medians,一个线性查找算法,关于它的具体介绍可以自行wiki,最近重温几个常见算法,发现好像没有用Python实现的BFPRT算法,所以就自己写了一个。这里算法最终结果是输出数组中第k大的数。</span>

算法步骤:

1. 将输入数组分为若干长度为5的一小组(若数组长度不够5整除,后面零散的可以忽略掉),排序获取中位数,把得到的中位数继续按照这种方法计算,得到最终的中位数,作为快速分割的主元pivot

2. 快速分割,按得到的中位数将数组分成两部分,小于pivot的和大于pivot的, 获取分割后pivot的下标index

3. 判断条件:若k=index,pivot即为输出结果,如若不等,判断k与index的大小关系,决定下一步操作是在小于pivot的部分好是大于pivot的部分(具体见代码)

def partition(arr, pivotIndex):
	index = 0
	r = len(arr) - 1
	pivotValue = arr[pivotIndex]
	arr[pivotIndex], arr[r] = arr[r], arr[pivotIndex]
	for i in range(0, r+1):
		if arr[i] < pivotValue:
			arr[index], arr[i] = arr[i], arr[index]
			index += 1
	arr[index], arr[r] = arr[r], arr[index]
	return index

def pivot_median(arr):
	n = len(arr)
	while n > 5:
		cols = n/5
		m = []
		for i in range(0, cols):
			s = sorted(arr[5*i:(5*i+5)])
			m.append(s[2])
		arr = m
		n = len(arr)
	arr.sort()
	return arr[n/2]

def bfprt(arr, k):
	pivot = pivot_median(arr)
	pivotIndex = arr.index(pivot)
	index = partition(arr, pivotIndex)
	n = len(arr)
	if k < n - index:
		return bfprt(arr[index+1:n], k)
	elif k == n - index:
		return pivot
	elif k > n - index:
		return bfprt(arr[0:index], k-(n-index))
能运行之后就没怎么修改代码,所以代码显得有些粗糙。细心的朋友会发现有一个问题就是对于重复出现的数字怎么办,这里没做什么处理,留给各位大神吧 偷笑

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值