首页 > 代码库 > 选择第n小的元素之python实现源码
选择第n小的元素之python实现源码
def partition(A, p, r): j = p+1 for i in range(p+1, r+1): if(A[i] < A[p]): tmp = A[i] A[i] = A[j] A[j] = tmp j = j + 1 tmp = A[p] A[p] = A[j-1] A[j-1] = tmp return j - 1 def select(A, p, r, i): if(p == r): return A[p] q = partition(A, p, r) #print q k = q - p + 1 if(k == i): return A[q] if(i < k): return select(A, p, q - 1, i) else: return select(A, q + 1, r, i - k)#test case#import pdb#pdb.set_trace()old = [2, 5, 3, 0, 7, 9, 6, 3] q = select(old, 0, 7, 6)print q
和快排一样使用partition对集合进行划分,在划分后的子集中查找第i小的元素。
时间复杂度是O(N)。
选择第n小的元素之python实现源码
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。