python查找第k小元素代码分享


Posted in Python onDecember 18, 2013
# -*- coding: utf-8 -*-
from random import randint
from math import ceil, floor
def _partition(A, l, r, i):
    """以A[i]为主元划分数组A[l..r],使得:
    A[l..m-1] <= A[m] < A[m+1..r]
    """
    A[i], A[r] = A[r], A[i] # i交换到末位r,作为主元
    pivot = A[r] # 主元
    m = l # 索引标记
    for n in xrange(l, r): # l..r-1
        if A[n] <= pivot:
            A[m], A[n] = A[n], A[m] # 交换
            m += 1 # 后移
    A[m], A[r] = A[r], A[m] # 主元到m位
    return m
def _rand(A, l, r):
    """随机划分主元"""
    return randint(l, r) # A[l..r]随机取一个
def _select(A, l, r, k, pivot_selector = _rand):
    """利用快排,得A[l..r]中第k小的数,k in [l+1,r+1]:
    其尾递归方式,伪码如下:
    SELECT(A, l, r, k)
    1  while true:
    2    i ← ? // 划分主元位置
    3    m ← PARTITION(A, l, r, i) // 数组划分
    4    n ← m - l + 1 // A[l..m]元素个数
    5    if k = n // 检查A[m]是否是第k小的元素
    6      then return A[m]
    7    elseif k < n // 左划分区
    8      r = m - 1
    9    else // 右划分区
    10     k = k - n
    11     l = m + 1
    Args:
        pivot_selector(Function): 主元选取方法,默认随机方式
    """
    if not A:
        return None
    if l == r:
        return A[l]
    while True:
        i = pivot_selector(A, l, r)
        m = _partition(A, l, r, i)
        n = m - l + 1
        if k == n:
            return A[m]
        elif k < n:
            r = m - 1
        else:
            k = k - n
            l = m + 1
def rand_select(A, k):
    """默认随机划分主元方式,k in [1, len(A)]
    E[T(n)] = O(n)
    """
    return _select(A, 0, len(A) - 1, k);

def _median(A, l, r):
    """对A[l..r]插入排序(原地)后选取其中位数位置"""
    for j in xrange(l, r + 1):
        k = A[j]
        i = j
        while i > l and A[i-1] > k:
            A[i] = A[i-1]
            i -= 1
        A[i] = k
    return l + int((r - l) * 0.5) # 下中位数
def _medianOfMedians(A, l, r):
    """中位数的中位数方式:
    1. 划分为floor(n/5)个5元组,剩下(n%5)组成最后一组。
    2. 找出ceil(n/5)个组各自的中位数。先对每组插入排序,再从中选出中位数。
    3. 对第2步中找出的ceil(n/5)个中位数重复上述操作,直到仅有一个中位数。
    """
    if l == r:
        return l
    n = r - l + 1 # 元素个数
    m = int(ceil(n / 5.0)) # 划分组数,每组5个元素
    for i in xrange(m):
        # 每组起始位和结束位
        sub_l = l + i * 5
        sub_r = sub_l + 4
        if sub_r > r:
            sub_r = r
        # 对每组元素插入排序后,选取中位数
        sub_m = _median(A, sub_l, sub_r) # 中位数索引
        # 交换中位数到前几位
        j = l + i
        A[j], A[sub_m] = A[sub_m], A[j]
    return _medianOfMedians(A, l, l + m - 1) # 中位数的中位数
def bfprt_select(A, k):
    """中位数的中位数方式(BFPRT算法)
    T(n) = O(n)
    """
    return _select(A, 0, len(A) - 1, k, _medianOfMedians);

def _median3(A, l, r):
    """三数中位数方式,取l,r,(l+r)/2三数中位数"""
    c = (l + r) / 2
    keys = [l, c, r]
    i = _median(keys, 0, 2)
    return keys[i]
def median_select(A, k):
    """三数中位数方式,以消除最坏情况"""
    return _select(A, 0, len(A) - 1, k, _median3);

if __name__ == '__main__':
    import random, time
    from copy import copy
    print('preparing data...')
    n = 1000000
    nums = range(n)
    random.shuffle(nums)
    print('ready go!')
    def timeit(fnc, *args, **kargs):
        print('%s starts processing' % fnc.__name__)
        begtime = time.clock()
        retval = fnc(*args, **kargs)
        endtime = time.clock()
        print('%s takes time : %f' % (fnc.__name__, endtime - begtime))
        return retval
    test_methods = [rand_select, bfprt_select, median_select]
    k = random.randrange(n) + 1
    dashes = '---' * 10
    for test in test_methods:
        print(dashes)
        nums_new = copy(nums)
        result = timeit(test, nums_new, k)
        print('the %dth smallest element: %d' % (k, result))
Python 相关文章推荐
举例讲解Python中装饰器的用法
Apr 27 Python
python遍历 truple list dictionary的几种方法总结
Sep 11 Python
Python selenium 三种等待方式详解(必会)
Sep 15 Python
13个最常用的Python深度学习库介绍
Oct 28 Python
Python字符串格式化%s%d%f详解
Feb 02 Python
Python 中Pickle库的使用详解
Feb 24 Python
Python 创建空的list,以及append用法讲解
May 04 Python
python+ffmpeg批量去视频开头的方法
Jan 09 Python
python 缺失值处理的方法(Imputation)
Jul 02 Python
django使用xadmin的全局配置详解
Nov 15 Python
Python实现bilibili时间长度查询的示例代码
Jan 14 Python
如何使用python包中的sched事件调度器
Apr 30 Python
python获取beautifulphoto随机某图片代码实例
Dec 18 #Python
python使用urllib2模块获取gravatar头像实例
Dec 18 #Python
python2.7删除文件夹和删除文件代码实例
Dec 18 #Python
python使用xmlrpc实例讲解
Dec 17 #Python
python三元运算符实现方法
Dec 17 #Python
用python写asp详细讲解
Dec 16 #Python
python模块restful使用方法实例
Dec 10 #Python
You might like
PHP中通过fopen()函数访问远程文件示例
2014/11/18 PHP
WordPress迁移时一些常见问题的解决方法整理
2015/11/24 PHP
让你的PHP7更快之Hugepage用法分析
2016/05/31 PHP
Node.js node-schedule定时任务隔多少分钟执行一次的方法
2015/02/10 Javascript
JS+CSS实现简单的二级下拉导航菜单效果
2015/09/21 Javascript
第四篇Bootstrap网格系统偏移列和嵌套列
2016/06/21 Javascript
微信小程序 css使用技巧总结
2017/01/09 Javascript
AngularJS实现的回到顶部指令功能实例
2017/05/17 Javascript
理解 JavaScript EventEmitter
2018/03/29 Javascript
node使用promise替代回调函数
2018/05/07 Javascript
微信小程序地图导航功能实现完整源代码附效果图(推荐)
2019/04/28 Javascript
react实现antd线上主题动态切换功能
2019/08/12 Javascript
深入解析微信小程序开发中遇到的几个小问题
2020/07/11 Javascript
toString.call()通用的判断数据类型方法示例
2020/08/28 Javascript
Vue时间轴 vue-light-timeline的用法说明
2020/10/29 Javascript
Python中exit、return、sys.exit()等使用实例和区别
2015/05/28 Python
Python读取和处理文件后缀为.sqlite的数据文件(实例讲解)
2017/06/27 Python
python实现将excel文件转化成CSV格式
2018/03/22 Python
Python3爬虫学习入门教程
2018/12/11 Python
PyQt5 实现字体大小自适应分辨率的方法
2019/06/18 Python
Matplotlib使用Cursor实现UI定位的示例代码
2020/03/12 Python
python一些性能分析的技巧
2020/08/30 Python
全球最大的网上自行车商店:Chain Reaction Cycles
2016/12/02 全球购物
Tory Burch德国官网:美国时尚生活品牌
2018/01/03 全球购物
手工制作的意大利皮革运动鞋:KOIO
2020/01/05 全球购物
加拿大专业美发产品购物网站:Chatters
2021/02/28 全球购物
Belvilla法国:休闲度假房屋出租
2020/10/03 全球购物
如何向接受结构参数的函数传入常数值
2016/02/17 面试题
暑期社会实践学生的自我评价
2014/01/09 职场文书
汽车销售经理岗位职责
2014/06/09 职场文书
2014离婚协议书范文
2014/09/10 职场文书
总经理检讨书
2014/09/15 职场文书
工作态度不端正检讨书
2014/10/04 职场文书
煤矿百日安全活动总结
2015/05/07 职场文书
2016幼儿教师自荐信范文
2016/01/28 职场文书
解决go在函数退出后子协程的退出问题
2021/04/30 Golang