-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathpriorityQueue.py
40 lines (29 loc) · 880 Bytes
/
priorityQueue.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import heapq
class PriorityQueue:
def __init__(self):
self.arr = []
self.index = 0
def isEmpty(self):
return len(self.arr) == 0
def enQueue(self, element, priority):
heapq.heappush(self.arr, (priority, self.index, element))
self.index += 1
def deQueue(self):
heapq.heappop(self.arr)[-1]
def printAll(self):
print(self.arr)
def getHighestPriority(self):
print(heapq.nsmallest(1, self.arr))
queue_instance = PriorityQueue()
queue_instance.enQueue('a', 1)
queue_instance.enQueue('aaa', 1)
queue_instance.enQueue('b', 3)
queue_instance.enQueue('c', 2)
queue_instance.enQueue('d', 5)
queue_instance.enQueue('e', 4)
queue_instance.printAll()
queue_instance.getHighestPriority()
queue_instance.deQueue()
queue_instance.deQueue()
queue_instance.printAll()
queue_instance.getHighestPriority()