def testExtract(self):
u"""测试堆的提取最值操作"""
heap = Heap([1, 2, 3, 4, 5, 6])
data = heap.extract()
self.assertEqual(1, data)
self.assertEqual(5, heap.size())
self.assertEqual([2, 4, 3, 6, 5], heap._heap)
data = heap.extract()
self.assertEqual(2, data)
self.assertEqual(4, heap.size())
self.assertEqual([3, 4, 5, 6], heap._heap)
def heapSort(data):
"""对给定的数组进行堆排序
Args:
data: 要排序的数组
Returns:
堆排序后的数组
"""
result = []
heap = Heap(data)
# 堆的根元素总是最小的，所以只要执行len(data)次提取最值（包括调整堆）即可
for i in xrange(len(data)):
result.append(heap.extract())
return result
def testInsert(self):
u"""测试堆的插入操作"""
heap = Heap([4, 6, 5, 7])
heap.insert(3)
self.assertEqual(5, heap.size())
self.assertEqual([3, 4, 5, 7, 6], heap._heap)
heap.insert(1)
self.assertEqual(6, heap.size())
self.assertEqual([1, 4, 3, 7, 6, 5], heap._heap)
def testInsert(self):
u"""测试堆的插入操作"""
heap = Heap([4, 6, 5, 7])
heap.insert(3)
self.assertEqual(5, heap.size())
self.assertEqual([3, 4, 5, 7, 6], heap._heap)
heap.insert(1)
self.assertEqual(6, heap.size())
self.assertEqual([1, 4, 3, 7, 6, 5], heap._heap)
def testBubbleDown(self):
u"""测试堆的向下冒泡调整操作"""
heap = Heap()
heap._heap = [3, 2, 4, 5, 1]
# 调整第2个元素
heap._bubbleDown(1)
self.assertEqual([3, 1, 4, 5, 2], heap._heap)
# 调整根元素
heap._bubbleDown(0)
self.assertEqual([1, 2, 4, 5, 3], heap._heap)
def testBubbleDown(self):
u"""测试堆的向下冒泡调整操作"""
heap = Heap()
heap._heap = [3, 2, 4, 5, 1]
# 调整第2个元素
heap._bubbleDown(1)
self.assertEqual([3, 1, 4, 5, 2], heap._heap)
# 调整根元素
heap._bubbleDown(0)
self.assertEqual([1, 2, 4, 5, 3], heap._heap)
def testExtract(self):
u"""测试堆的提取最值操作"""
heap = Heap([1, 2, 3, 4, 5, 6])
data = heap.extract()
self.assertEqual(1, data)
self.assertEqual(5, heap.size())
self.assertEqual([2, 4, 3, 6, 5], heap._heap)
data = heap.extract()
self.assertEqual(2, data)
self.assertEqual(4, heap.size())
self.assertEqual([3, 4, 5, 6], heap._heap)
def testBuildHeap(self):
u"""测试堆的初始化并建立最小堆"""
heap = Heap([3, 2, 6, 4, 1, 5])
self.assertEqual([1, 2, 5, 4, 3, 6], heap._heap)
def testSize(self):
u"""测试堆的获取尺寸方法"""
heap = Heap(range(5))
self.assertEqual(5, heap.size())
def testSize(self):
u"""测试堆的获取尺寸方法"""
heap = Heap(range(5))
self.assertEqual(5, heap.size())