排序算法集合（1）-Python实现的堆排序

      最近自己动手实现了各种排序算法，有冒泡、插入、快速排序、二叉树排序等等，觉得快速排序的效率（对于一般的乱序而言）的确无可匹敌。当然，堆排序也很酷。不过前几个使用c#实现的，而堆排序是用python实现的，没办法做benchmark了。今后几天陆续贴出来，供大家参看。

      堆排序的算法请参考：http://baike.baidu.com/view/157305.htm

      其中，createHeap()能够将一个list创建成堆（最大值在最上），而它是基于insertIntoHeap()的，后者的原理是把一个值插入到一个堆中。还有一个重要的函数是topDownHeap(), 这是堆排序的关键一步，将位于堆队列最后的元素，放置到堆顶，然后通过向左右分支做比较，将其沉降到一个合适位置。最后，还有一个辅助功能printHeap(),它能将一个堆队列，以树状打印出来。

 

---

# heapSort.py import random def insertIntoHeap(x,regHeap): if regHeap==None: regHeap=[x] return regHeap; regHeap.append(x) #insert to last pos=len(regHeap) while True: if pos==1: return regHeap fatherPos=pos/2 if regHeap[fatherPos-1]<regHeap[pos-1]: regHeap[fatherPos-1],regHeap[pos-1]=regHeap[pos-1],regHeap[fatherPos-1] pos=fatherPos else: return regHeap def printHeap(regHeap): if regHeap==None: print "Empty" i=1 #height from root while True: if i*2>=len(regHeap): print regHeap[i-1:len(regHeap)] return else: print regHeap[i-1:2*i-1] i*=2 def createHeap(a): b=[] for i in a: insertIntoHeap(i,b) return b def topDownHeap(a): if a==None or type(a)!=list: return if a==[]or len(a)==1: return a pos=0 while True: if (pos+1)*2>len(a): #Node[pos] has no children return a if (pos+1)*2==len(a): #Node[pos] has only left child if a[pos]<a[2*pos+1]: a[pos],a[2*pos+1]=a[2*pos+1],a[pos] return a if (pos+1)*2+1<=len(a): #Node[pos] has two children if a[pos]<a[2*pos+1]and a[2*pos+1]>=a[2*pos+2]: a[pos],a[2*pos+1]=a[2*pos+1],a[pos] pos=2*pos+1 elif a[pos]<a[2*pos+2]and a[2*pos+1]<a[2*pos+2]: a[pos],a[2*pos+2]=a[2*pos+2],a[pos] pos=2*pos+2 else: return a #Do not need to adjust def heapSort(a): if a==None or type(a)!=list: return #init b=createHeap(a) a=[] while b!=[]: a.append(b[0]) b[0]=b[len(b)-1] b.pop() b=topDownHeap(b) return a #main,a sorting example a=[] for i in range(30): a.append(random.randint(1, 400)) print a print heapSort(a)