python計(jì)算最大優(yōu)先級(jí)隊(duì)列實(shí)例
# -*- coding: utf-8 -*-
class Heap(object):
@classmethod
def parent(cls, i):
"""父結(jié)點(diǎn)下標(biāo)"""
return int((i - 1) >> 1);
@classmethod
def left(cls, i):
"""左兒子下標(biāo)"""
return (i << 1) + 1;
@classmethod
def right(cls, i):
"""右兒子下標(biāo)"""
return (i << 1) + 2;
class MaxPriorityQueue(list, Heap):
@classmethod
def max_heapify(cls, A, i, heap_size):
"""最大堆化A[i]為根的子樹(shù)"""
l, r = cls.left(i), cls.right(i)
if l < heap_size and A[l] > A[i]:
largest = l
else:
largest = i
if r < heap_size and A[r] > A[largest]:
largest = r
if largest != i:
A[i], A[largest] = A[largest], A[i]
cls.max_heapify(A, largest, heap_size)
def maximum(self):
"""返回最大元素,偽碼如下:
HEAP-MAXIMUM(S)
1 return A[1]
T(n) = O(1)
"""
return self[0]
def extract_max(self):
"""去除并返回最大元素,偽碼如下:
HEAP-EXTRACT-MAX(A)
1 if heap-size[A] < 1
2 then error "heap underflow"
3 max ← A[1]
4 A[1] ← A[heap-size[A]] // 尾元素放到第一位
5 heap-size[A] ← heap-size[A] - 1 // 減小heap-size[A]
6 MAX-HEAPIFY(A, 1) // 保持最大堆性質(zhì)
7 return max
T(n) = θ(lgn)
"""
heap_size = len(self)
assert heap_size > 0, "heap underflow"
val = self[0]
tail = heap_size - 1
self[0] = self[tail]
self.max_heapify(self, 0, tail)
self.pop(tail)
return val
def increase_key(self, i, key):
"""將i處的值增加到key,偽碼如下:
HEAP-INCREASE-KEY(A, i, key)
1 if key < A[i]
2 the error "new key is smaller than current key"
3 A[i] ← key
4 while i > 1 and A[PARENT(i)] < A[i] // 不是根結(jié)點(diǎn)且父結(jié)點(diǎn)更小時(shí)
5 do exchange A[i] ↔ A[PARENT(i)] // 交換兩元素
6 i ← PARENT(i) // 指向父結(jié)點(diǎn)位置
T(n) = θ(lgn)
"""
val = self[i]
assert key >= val, "new key is smaller than current key"
self[i] = key
parent = self.parent
while i > 0 and self[parent(i)] < self[i]:
self[i], self[parent(i)] = self[parent(i)], self[i]
i = parent(i)
def insert(self, key):
"""將key插入A,偽碼如下:
MAX-HEAP-INSERT(A, key)
1 heap-size[A] ← heap-size[A] + 1 // 對(duì)元素個(gè)數(shù)增加
2 A[heap-size[A]] ← -∞ // 初始新增加元素為-∞
3 HEAP-INCREASE-KEY(A, heap-size[A], key) // 將新增元素增加到key
T(n) = θ(lgn)
"""
self.append(float('-inf'))
self.increase_key(len(self) - 1, key)
if __name__ == '__main__':
import random
keys = range(10)
random.shuffle(keys)
print(keys)
queue = MaxPriorityQueue() # 插入方式建最大堆
for i in keys:
queue.insert(i)
print(queue)
print('*' * 30)
for i in range(len(keys)):
val = i % 3
if val == 0:
val = queue.extract_max() # 去除并返回最大元素
elif val == 1:
val = queue.maximum() # 返回最大元素
else:
val = queue[1] + 10
queue.increase_key(1, val) # queue[1]增加10
print(queue, val)
print([queue.extract_max() for i in range(len(queue))])
- Python中棧、隊(duì)列與優(yōu)先級(jí)隊(duì)列的實(shí)現(xiàn)方法
- Python利用heapq實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列的方法
- Python cookbook(數(shù)據(jù)結(jié)構(gòu)與算法)實(shí)現(xiàn)優(yōu)先級(jí)隊(duì)列的方法示例
- Python實(shí)現(xiàn)優(yōu)先級(jí)隊(duì)列結(jié)構(gòu)的方法詳解
- python計(jì)算最小優(yōu)先級(jí)隊(duì)列代碼分享
- Python實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列的方法
相關(guān)文章
Python隨機(jī)數(shù)random模塊使用指南
本文給大家分享的是Python隨機(jī)數(shù)random模塊的幾個(gè)常用的方法,非常的簡(jiǎn)單,小伙伴們喜歡的話,后續(xù)繼續(xù)深入探討2016-09-09Python實(shí)現(xiàn)解析命令行參數(shù)的常見(jiàn)方法總結(jié)
除ide的執(zhí)行方式外,命令行的方式執(zhí)行Python腳本是參數(shù)化程序執(zhí)行的一種常見(jiàn)且簡(jiǎn)單的方法。本文總結(jié)了三個(gè)常見(jiàn)的獲取和解析命令行參數(shù)的方法,需要的可以參考一下2022-10-10python爬蟲(chóng)入門(mén)教程--優(yōu)雅的HTTP庫(kù)requests(二)
requests 實(shí)現(xiàn)了 HTTP 協(xié)議中絕大部分功能,它提供的功能包括 Keep-Alive、連接池、Cookie持久化、內(nèi)容自動(dòng)解壓、HTTP代理、SSL認(rèn)證等很多特性,下面這篇文章主要給大家介紹了python爬蟲(chóng)入門(mén)中關(guān)于優(yōu)雅的HTTP庫(kù)requests的相關(guān)資料,需要的朋友可以參考下。2017-05-05JPype實(shí)現(xiàn)在python中調(diào)用JAVA的實(shí)例
本篇文章主要介紹了JPype實(shí)現(xiàn)在python中調(diào)用JAVA的實(shí)例,小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-07-07