对于内置对象,python都提供了print方法,直接输出元素。
自定义对象,一定要提供__str__方法,并且返回一个string对象,才可以支持print方法。
binaryHeap的str方法:
class BinHeap:
def __init__(self):
self.heapList = [0]
self.currentSize = 0
self.current = 0
def len(self):
return len(self.heapList) - 1
def __str__(self):
#
s1=''
for c in self:
s1+=' '+str(c)
#
return s1
def __iter__(self):
return self
def __next__(self):
#k = 1
if self.current == self.currentSize:
raise StopIteration
else:
self.current += 1
return self.heapList[self.current]
因为目标是输出全部元素,所以iter next需要一并实现