Total Complexity | 20 |
Total Lines | 64 |
Duplicated Lines | 0 % |
Changes | 1 | ||
Bugs | 0 | Features | 0 |
1 | def exchange(a, i, j): |
||
7 | class MinPQ(object): |
||
8 | s = None |
||
9 | N = 0 |
||
10 | compare = None |
||
11 | |||
12 | def __init__(self, compare=None): |
||
13 | self.s = [0] * 10 |
||
14 | self.N = 0 |
||
15 | if compare is None: |
||
16 | compare = lambda x, y: x - y |
||
17 | self.compare = compare |
||
18 | |||
19 | def less(self, a1, a2): |
||
20 | return self.compare(a1, a2) < 0 |
||
21 | |||
22 | def enqueue(self, item): |
||
23 | if self.N + 1 >= len(self.s): |
||
24 | self.resize(len(self.s) * 2) |
||
25 | |||
26 | self.N += 1 |
||
27 | self.s[self.N] = item |
||
28 | self.swim(self.N) |
||
29 | |||
30 | def resize(self, new_length): |
||
31 | temp = [0] * new_length |
||
32 | length = min(new_length, len(self.s)) |
||
33 | for i in range(length): |
||
34 | temp[i] = self.s[i] |
||
35 | self.s = temp |
||
36 | |||
37 | def swim(self, k): |
||
38 | while k > 1: |
||
39 | parent = k // 2 |
||
40 | if self.less(self.s[k], self.s[parent]): |
||
41 | exchange(self.s, k, parent) |
||
42 | k = parent |
||
43 | else: |
||
44 | break |
||
45 | |||
46 | def del_min(self): |
||
47 | if self.N == 0: |
||
48 | return None |
||
49 | item = self.s[1] |
||
50 | exchange(self.s, 1, self.N) |
||
51 | self.N -= 1 |
||
52 | self.sink(1) |
||
53 | return item |
||
54 | |||
55 | def sink(self, k): |
||
56 | while 2 * k <= self.N: |
||
57 | child = k * 2 |
||
58 | if child < self.N and self.less(self.s[child + 1], self.s[child]): |
||
59 | child += 1 |
||
60 | if self.less(self.s[child], self.s[k]): |
||
61 | exchange(self.s, k, child) |
||
62 | k = child |
||
63 | else: |
||
64 | break |
||
65 | |||
66 | def size(self): |
||
67 | return self.N |
||
68 | |||
69 | def is_empty(self): |
||
70 | return self.N == 0 |
||
71 |