Python数据结构与算法-在M个数中找K个最小的数

'''
查找最小的k个元素
题目：输入n个整数，输出其中最小的k个。
例如输入1，2，3，4，5，6，7和8这8个数字，则最小的4个数字为1，2，3和4
'''

def adjustHeap(heap, page):
    '''
    堆的调整
    param:heap是堆的数组，page是需要调整的节点
    '''
    temp = page
    #找page和page左节点和右节点三个点中，最大的一个点
    if 2*temp+1 < len(heap):
        if heap[2*temp+1] > heap[temp]:
           temp = 2*temp+1
    if 2*temp+2 < len(heap):
        if heap[2*temp+2] > heap[temp]:
            temp = 2*temp+2
    #调整用最大的节点作为根节点，然后在对调整后的节点继续做调整
    if temp != page:
        temp_data = heap[page]
        heap[page] = heap[temp]
        heap[temp] = temp_data
        adjustHeap(heap, temp)


if __name__ == "__main__":   
    m = [10,-9,0,100,90,1,4,-9]
    k = 7
    heap = []
    len_m = len(m)
    #对异常输入做返回
    if k <=0 or len_m<=0 or k > len_m:
        print None
    #如果长度相当直接返回
    else:
        if k==len_m:
            print m

        else:
            #否则先对前k个节点，从后向前调整
            heap = m[:k]
            index = (k-1)/2
            while index >= 0:
                adjustHeap(heap, index)
                index = index - 1
            tail = k-len_m
            m = m[tail:]
            #然后在用后面的数据跟根节点进行大小比较，如果比根节点小，则替换根节点，在从上向下做调整
            for i in m:
                if i < heap[0]:
                    heap[0] = i
                    adjustHeap(heap, 0)
            print heap