亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定

[碩.Love Python] Heap(堆)

標簽:
Python
class MinHeap(object):
    def __init__(self, iterable=()):
        self.array = [None]
        self.array.extend(iterable)
        self.build()

    def build(self):
        a, size = self.array, self.size()
        for i in xrange(size / 2, 0, -1):
            c = 2 * i
            while c <= size:
                if c + 1 <= size and a[c+1] < a[c]:
                    c += 1

                if a[c] > a[c/2]:
                    break

                a[c], a[c/2] = a[c/2], a[c]
                c *= 2

    def insert(self, k):
        self.array.append(k)
        a, i = self.array, self.size()

        while i > 1 and k < a[i/2]:
            a[i] = a[i/2]
            i /= 2

        a[i] = k

    def pop(self):
        size = self.size()
        if size == 0:
            raise IndexError('pop from empty heap')

        a = self.array
        res = a[1]
        last = a.pop()
        size -= 1

        if size > 0:
            c = 2 * 1

            while c <= size:
                if c + 1 <= size and a[c+1] < a[c]:
                    c += 1

                if a[c] >= last:
                    break

                a[c/2] = a[c]
                c *= 2

            a[c/2] = last

        return res

    def size(self):
        return len(self.array) - 1

if __name__ == '__main__':
    from random import shuffle
    data = range(50)
    shuffle(data)
    print data
    h = MinHeap(data)

    for _ in xrange(len(data)):
         print h.pop()
點擊查看更多內容
4人點贊

若覺得本文不錯,就分享一下吧!

評論

作者其他優質文章

正在加載中
感謝您的支持,我會繼續努力的~
掃碼打賞,你說多少就多少
贊賞金額會直接到老師賬戶
支付方式
打開微信掃一掃,即可進行掃碼打賞哦
今天注冊有機會得

100積分直接送

付費專欄免費學

大額優惠券免費領

立即參與 放棄機會
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號

舉報

0/150
提交
取消