from random import randint
def swap(array:[], index1, index2):
array[index1], array[index2] = array[index2], array[index1]
def partition(array:[], startIndex:int, endIndex:int) -> int:
pivot = array[endIndex]
lastLessorIndex = startIndex - 1
for i in range(startIndex, endIndex):
if array[i] <= pivot:
lastLessorIndex += 1
swap(array, i, lastLessorIndex)
lastLessorIndex += 1
swap(array, endIndex, lastLessorIndex)
return lastLessorIndex
def randomizedPartition(array:[], startIndex:int, endIndex:int) -> int:
swap(array, endIndex, randint(startIndex, endIndex))
return partition(array, startIndex, endIndex)
def randomizedSelect(array:[], startIndex:int, endIndex:int, targetOrder:int) -> int:
# targetOrder that begins with 0 serves more like an index.
if startIndex == endIndex:
return array[startIndex]
currentIndex = randomizedPartition(array, startIndex, endIndex)
targetIndex = startIndex + targetOrder
if currentIndex == targetIndex:
return array[currentIndex]
elif currentIndex < targetIndex:
return randomizedSelect(array, currentIndex+1, endIndex, \
targetIndex-currentIndex-1)
else:
return randomizedSelect(array, startIndex, currentIndex-1, targetOrder)
Randomized Select. Expected Θ(n).
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
推荐阅读更多精彩内容
- Effectively, factoring out the selectMedian function is q...
- 在SQL Server里面有top关键字可以很方便的取出前N条记录,但是Oracle里面却没有top的使用。解决方...
- Consider sorting n numbers stored in array A by first fin...