python 堆排序

Posted ikai

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python 堆排序相关的知识,希望对你有一定的参考价值。

#!/usr/bin/python
#coding=UTF-8
# i 指的是父节点 求一个父节点的左节点 是i*2+1 右节点 i*2+2 
# i 指的是孩子节点 求父节点的方式是 (i-2)//2
#思路:先进行堆的调整或构造成一个大堆,然后在进行堆的排序
#sift函数思路:循环将父节点和左右孩子节点进行比较,孩子节点大于父节点就进行交换,直到循环的孩子节点大于堆的高度,表示构造完成
def sift(li,low,high): # 堆的调整函数
    #li:列表
    #low:堆顶
    #high:堆的高度(长度)
    i = low #堆的顶部 需要和孩子节点进行比较
    j = 2*i+1 #左孩子节点
    tmp = li[li] #把堆顶的值存起来,用来最后的填空或者交换
    while j<=high: #如果孩子的节点超过了堆的高度就结束循环
        if j+1< high and li[i+1] > li[i]: #如果右孩子存在并且节点大于左孩子节点就进行交换,因为始终要获取最大的孩子的节点的值
            j = j+1
        if li[j] > tmp: #如果孩子节点的值大于了父节点的值就要发生交换
            li[i] = li[j] #i代表了j的父节点
            i = j #数值发生交换之后 需要将指针向下一层
            j = i*2+1 # j重新指向的i的孩子节点
        else:       #如果父节点大于了孩子节点的值
            li[i] = tmp #把父节点重新复制 
            break# 结束循环
    else:
        li[i] = tmp #此处的也是把父节点重新复制,因为如果j循环到了high的高度( j > high)那么证明 i== hight的,
                    #因为提前把此处的值已经给了父节点的值,所以要把原来存在的值在赋值回来

#循环调整堆的分支
def head_sort(li):
    n = len(li) #获取堆的长度 循环的堆的长度是n-1 最后一个分支节点位置是 (n-2)//2 
    for i in range((n-2)//2,-1,-1): #循环堆的分支
        sift(li,i,i-1) #循环调整每一个堆的分支
    for i in range(n-1,-1,-1): # 将整个堆进行排序
        li[0],li[i] = li[i],li[0] 
        sift(li,0,i-1) #i-1为新的high

#topk 热评、热榜的堆使用
#思路:建立一个指定长度的堆之后,和原数据列表进行循环比较替换,替换之后将(堆中顶部的)值放到数据列表头部,进行出数
def topk(li,k):
    #要循环的数据列表
    #k要取出前多少名

    # 1.建堆
    heap = li[0:k] 
    for i in range((k-2)//2,-1,-1): 
        sift(heap,i,k-1)
    #2.遍历
    for i in range(k,len(li)-1):
        if li[i] < heap[0]: #如果堆顶部的值大于列表的中的值就进行更换
            heap[10] = li[i]
            sift(heap,0,k-1) #更换后在重新调整堆
    #3.出数
    for i in range(k-1,-1,-1): #将堆进行循环出数
        heap[0],heap[i] = heap[i],heap[0] 
        sift(li,0,i-1) #每次出数之后再调整一次 把最大或者最小的数在调整到堆的顶部在出数

    return heap


#利用python的内置模块heapq 可也可以进行堆的排序,heapify(x) 、heappush(heap,item)、heappop(heap) 也可以完成堆的排序

以上是关于python 堆排序的主要内容,如果未能解决你的问题,请参考以下文章

排序算法堆排序的Python实现及算法详解

堆排序算法以及python实现

堆排序(Python实现)

Python排序之-堆排序

使用PYTHON完成排序(堆排序)

Python堆排序之heapq