python最大堆的实现且判断是否为最大堆

class Heap:
    def __init__(self):
        self.data=[]

    @staticmethod
    def _parent(idx):
        return (idx-1)//2


    @staticmethod
    def _left(idx):
        return idx*2+1


    @staticmethod
    def _right(idx):
        return idx*2+2


    def max(self):
        return self.data[0]

    def add(self,x):
        self.data.append(x)
        i=len(self.data)-1
        while i>0:
            p=Heap._parent(i)
            if self.data[i]>self.data[p]:
                self.data[i],self.data[p]=self.data[p],self.data[i]
                i=p
            else:
                break


    def pop_max(self):
        m=self.data[0]
        self.data[0]=self.data[-1]
        del self.data[-1]
        i=0
        while i<len(self.data):
            l=Heap._right(i)
            r=Heap._left(i)
            mi=i
            if l<len(self.data) and self.data[l]>self.data[i]:
                mi=l
            if r<len(self.data) and self.data[r]>self.data[mi]:
                mi=r
            if mi!=i:
                self.data[i],self.data[mi]=self.data[mi],self.data[i]
                i=mi
            else:
                break
        return m



    def __repr__(self):
        return repr(self.data)



    def __bool__(self):
        return len(self.data)>0

判断是否为最大堆

def judgeIsMaxHeap(heapList):
    heapLenght=len(heapList)
    if heapList[0]>heapList[-1]:
        for i in range((heapLenght//2)):
            if (2*i+2)>=heapLenght:
                if heapList[i]>=heapList[2*i+1]:
                    print(i)
                    continue
                else:
                    return False
            elif heapList[i]>=heapList[2*i+1] and heapList[i]>=heapList[2*i+2]:
                continue
            else:
                return False
        return True


heapList=[9, 8, 5, 6, 7, 3, 0, 1, 2, 4]
print(judgeIsMaxHeap(heapList))
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值