forked from aaronbloomfield/pdr
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbinary_heap.cpp.html
More file actions
123 lines (110 loc) · 14.1 KB
/
Copy pathbinary_heap.cpp.html
File metadata and controls
123 lines (110 loc) · 14.1 KB
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
<!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML//EN">
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=iso-8859-1">
<meta name="GENERATOR" content="GNU source-highlight 3.1.5
by Lorenzo Bettini
http://www.lorenzobettini.it
http://www.gnu.org/software/src-highlite">
<title>binary_heap.cpp</title>
</head>
<body bgcolor="white">
<pre><tt><i><font color="#9A1900">// Code written by Aaron Bloomfield, 2014</font></i>
<i><font color="#9A1900">// Released under a CC BY-SA license</font></i>
<i><font color="#9A1900">// This code is part of the https://github.com/aaronbloomfield/pdr repository</font></i>
<b><font color="#000080">#include</font></b> <font color="#FF0000"><iostream></font>
<b><font color="#000080">#include</font></b> <font color="#FF0000">"binary_heap.h"</font>
<b><font color="#0000FF">using</font></b> <b><font color="#0000FF">namespace</font></b> std<font color="#990000">;</font>
<i><font color="#9A1900">// default constructor</font></i>
binary_heap<font color="#990000">::</font><b><font color="#000000">binary_heap</font></b><font color="#990000">()</font> <font color="#990000">:</font> <b><font color="#000000">heap_size</font></b><font color="#990000">(</font><font color="#993399">0</font><font color="#990000">)</font> <font color="#FF0000">{</font>
heap<font color="#990000">.</font><b><font color="#000000">push_back</font></b><font color="#990000">(</font><font color="#993399">0</font><font color="#990000">);</font>
<font color="#FF0000">}</font>
<i><font color="#9A1900">// builds a heap from an unsorted vector</font></i>
binary_heap<font color="#990000">::</font><b><font color="#000000">binary_heap</font></b><font color="#990000">(</font><font color="#008080">vector<int></font> vec<font color="#990000">)</font> <font color="#990000">:</font> <b><font color="#000000">heap_size</font></b><font color="#990000">(</font>vec<font color="#990000">.</font><b><font color="#000000">size</font></b><font color="#990000">())</font> <font color="#FF0000">{</font>
heap <font color="#990000">=</font> vec<font color="#990000">;</font>
heap<font color="#990000">.</font><b><font color="#000000">push_back</font></b><font color="#990000">(</font>heap<font color="#990000">[</font><font color="#993399">0</font><font color="#990000">]);</font>
heap<font color="#990000">[</font><font color="#993399">0</font><font color="#990000">]</font> <font color="#990000">=</font> <font color="#993399">0</font><font color="#990000">;</font>
<b><font color="#0000FF">for</font></b> <font color="#990000">(</font> <font color="#009900">int</font> i <font color="#990000">=</font> heap_size<font color="#990000">/</font><font color="#993399">2</font><font color="#990000">;</font> i <font color="#990000">></font> <font color="#993399">0</font><font color="#990000">;</font> i<font color="#990000">--</font> <font color="#990000">)</font>
<b><font color="#000000">percolateDown</font></b><font color="#990000">(</font>i<font color="#990000">);</font>
<font color="#FF0000">}</font>
<i><font color="#9A1900">// the destructor doesn't need to do much</font></i>
binary_heap<font color="#990000">::~</font><b><font color="#000000">binary_heap</font></b><font color="#990000">()</font> <font color="#FF0000">{</font>
<font color="#FF0000">}</font>
<font color="#009900">void</font> binary_heap<font color="#990000">::</font><b><font color="#000000">insert</font></b><font color="#990000">(</font><font color="#009900">int</font> x<font color="#990000">)</font> <font color="#FF0000">{</font>
<i><font color="#9A1900">// a vector push_back will resize as necessary</font></i>
heap<font color="#990000">.</font><b><font color="#000000">push_back</font></b><font color="#990000">(</font>x<font color="#990000">);</font>
<i><font color="#9A1900">// move it up into the right position</font></i>
<b><font color="#000000">percolateUp</font></b><font color="#990000">(++</font>heap_size<font color="#990000">);</font>
<font color="#FF0000">}</font>
<font color="#009900">void</font> binary_heap<font color="#990000">::</font><b><font color="#000000">percolateUp</font></b><font color="#990000">(</font><font color="#009900">int</font> hole<font color="#990000">)</font> <font color="#FF0000">{</font>
<i><font color="#9A1900">// get the value just inserted</font></i>
<font color="#009900">int</font> x <font color="#990000">=</font> heap<font color="#990000">[</font>hole<font color="#990000">];</font>
<i><font color="#9A1900">// while we haven't run off the top and while the</font></i>
<i><font color="#9A1900">// value is less than the parent...</font></i>
<b><font color="#0000FF">for</font></b> <font color="#990000">(</font> <font color="#990000">;</font> <font color="#990000">(</font>hole <font color="#990000">></font> <font color="#993399">1</font><font color="#990000">)</font> <font color="#990000">&&</font> <font color="#990000">(</font>x <font color="#990000"><</font> heap<font color="#990000">[</font>hole<font color="#990000">/</font><font color="#993399">2</font><font color="#990000">]);</font> hole <font color="#990000">/=</font> <font color="#993399">2</font> <font color="#990000">)</font>
heap<font color="#990000">[</font>hole<font color="#990000">]</font> <font color="#990000">=</font> heap<font color="#990000">[</font>hole<font color="#990000">/</font><font color="#993399">2</font><font color="#990000">];</font> <i><font color="#9A1900">// move the parent down</font></i>
<i><font color="#9A1900">// correct position found! insert at that spot</font></i>
heap<font color="#990000">[</font>hole<font color="#990000">]</font> <font color="#990000">=</font> x<font color="#990000">;</font>
<font color="#FF0000">}</font>
<font color="#009900">int</font> binary_heap<font color="#990000">::</font><b><font color="#000000">deleteMin</font></b><font color="#990000">()</font> <font color="#FF0000">{</font>
<i><font color="#9A1900">// make sure the heap is not empty</font></i>
<b><font color="#0000FF">if</font></b> <font color="#990000">(</font> heap_size <font color="#990000">==</font> <font color="#993399">0</font> <font color="#990000">)</font>
<b><font color="#0000FF">throw</font></b> <font color="#FF0000">"deleteMin() called on empty heap"</font><font color="#990000">;</font>
<i><font color="#9A1900">// save the value to be returned</font></i>
<font color="#009900">int</font> ret <font color="#990000">=</font> heap<font color="#990000">[</font><font color="#993399">1</font><font color="#990000">];</font>
<i><font color="#9A1900">// move the last inserted position into the root</font></i>
heap<font color="#990000">[</font><font color="#993399">1</font><font color="#990000">]</font> <font color="#990000">=</font> heap<font color="#990000">[</font>heap_size<font color="#990000">--];</font>
<i><font color="#9A1900">// make sure the vector knows that there is one less element</font></i>
heap<font color="#990000">.</font><b><font color="#000000">pop_back</font></b><font color="#990000">();</font>
<i><font color="#9A1900">// percolate the root down to the proper position</font></i>
<b><font color="#000000">percolateDown</font></b><font color="#990000">(</font><font color="#993399">1</font><font color="#990000">);</font>
<i><font color="#9A1900">// return the old root node</font></i>
<b><font color="#0000FF">return</font></b> ret<font color="#990000">;</font>
<font color="#FF0000">}</font>
<font color="#009900">void</font> binary_heap<font color="#990000">::</font><b><font color="#000000">percolateDown</font></b><font color="#990000">(</font><font color="#009900">int</font> hole<font color="#990000">)</font> <font color="#FF0000">{</font>
<i><font color="#9A1900">// get the value to percolate down</font></i>
<font color="#009900">int</font> x <font color="#990000">=</font> heap<font color="#990000">[</font>hole<font color="#990000">];</font>
<i><font color="#9A1900">// while there is a left child...</font></i>
<b><font color="#0000FF">while</font></b> <font color="#990000">(</font> hole<font color="#990000">*</font><font color="#993399">2</font> <font color="#990000"><=</font> heap_size <font color="#990000">)</font> <font color="#FF0000">{</font>
<font color="#009900">int</font> child <font color="#990000">=</font> hole<font color="#990000">*</font><font color="#993399">2</font><font color="#990000">;</font> <i><font color="#9A1900">// the left child</font></i>
<i><font color="#9A1900">// is there a right child? if so, is it lesser?</font></i>
<b><font color="#0000FF">if</font></b> <font color="#990000">(</font> <font color="#990000">(</font>child<font color="#990000">+</font><font color="#993399">1</font> <font color="#990000"><=</font> heap_size<font color="#990000">)</font> <font color="#990000">&&</font> <font color="#990000">(</font>heap<font color="#990000">[</font>child<font color="#990000">+</font><font color="#993399">1</font><font color="#990000">]</font> <font color="#990000"><</font> heap<font color="#990000">[</font>child<font color="#990000">])</font> <font color="#990000">)</font>
child<font color="#990000">++;</font>
<i><font color="#9A1900">// if the child is greater than the node...</font></i>
<b><font color="#0000FF">if</font></b> <font color="#990000">(</font> x <font color="#990000">></font> heap<font color="#990000">[</font>child<font color="#990000">]</font> <font color="#990000">)</font> <font color="#FF0000">{</font>
heap<font color="#990000">[</font>hole<font color="#990000">]</font> <font color="#990000">=</font> heap<font color="#990000">[</font>child<font color="#990000">];</font> <i><font color="#9A1900">// move child up</font></i>
hole <font color="#990000">=</font> child<font color="#990000">;</font> <i><font color="#9A1900">// move hole down</font></i>
<font color="#FF0000">}</font> <b><font color="#0000FF">else</font></b>
<b><font color="#0000FF">break</font></b><font color="#990000">;</font>
<font color="#FF0000">}</font>
<i><font color="#9A1900">// correct position found! insert at that spot</font></i>
heap<font color="#990000">[</font>hole<font color="#990000">]</font> <font color="#990000">=</font> x<font color="#990000">;</font>
<font color="#FF0000">}</font>
<font color="#009900">int</font> binary_heap<font color="#990000">::</font><b><font color="#000000">findMin</font></b><font color="#990000">()</font> <font color="#FF0000">{</font>
<b><font color="#0000FF">if</font></b> <font color="#990000">(</font> heap_size <font color="#990000">==</font> <font color="#993399">0</font> <font color="#990000">)</font>
<b><font color="#0000FF">throw</font></b> <font color="#FF0000">"findMin() called on empty heap"</font><font color="#990000">;</font>
<b><font color="#0000FF">return</font></b> heap<font color="#990000">[</font><font color="#993399">1</font><font color="#990000">];</font>
<font color="#FF0000">}</font>
<font color="#009900">unsigned</font> <font color="#009900">int</font> binary_heap<font color="#990000">::</font><b><font color="#000000">size</font></b><font color="#990000">()</font> <font color="#FF0000">{</font>
<b><font color="#0000FF">return</font></b> heap_size<font color="#990000">;</font>
<font color="#FF0000">}</font>
<font color="#009900">void</font> binary_heap<font color="#990000">::</font><b><font color="#000000">makeEmpty</font></b><font color="#990000">()</font> <font color="#FF0000">{</font>
heap_size <font color="#990000">=</font> <font color="#993399">0</font><font color="#990000">;</font>
<font color="#FF0000">}</font>
<font color="#009900">bool</font> binary_heap<font color="#990000">::</font><b><font color="#000000">isEmpty</font></b><font color="#990000">()</font> <font color="#FF0000">{</font>
<b><font color="#0000FF">return</font></b> heap_size <font color="#990000">==</font> <font color="#993399">0</font><font color="#990000">;</font>
<font color="#FF0000">}</font>
<font color="#009900">void</font> binary_heap<font color="#990000">::</font><b><font color="#000000">print</font></b><font color="#990000">()</font> <font color="#FF0000">{</font>
cout <font color="#990000"><<</font> <font color="#FF0000">"("</font> <font color="#990000"><<</font> heap<font color="#990000">[</font><font color="#993399">0</font><font color="#990000">]</font> <font color="#990000"><<</font> <font color="#FF0000">") "</font><font color="#990000">;</font>
<b><font color="#0000FF">for</font></b> <font color="#990000">(</font> <font color="#009900">int</font> i <font color="#990000">=</font> <font color="#993399">1</font><font color="#990000">;</font> i <font color="#990000"><=</font> heap_size<font color="#990000">;</font> i<font color="#990000">++</font> <font color="#990000">)</font> <font color="#FF0000">{</font>
cout <font color="#990000"><<</font> heap<font color="#990000">[</font>i<font color="#990000">]</font> <font color="#990000"><<</font> <font color="#FF0000">" "</font><font color="#990000">;</font>
<i><font color="#9A1900">// next line from from http://tinyurl.com/mf9tbgm</font></i>
<font color="#009900">bool</font> isPow2 <font color="#990000">=</font> <font color="#990000">(((</font>i<font color="#990000">+</font><font color="#993399">1</font><font color="#990000">)</font> <font color="#990000">&</font> <font color="#990000">~(</font>i<font color="#990000">))==(</font>i<font color="#990000">+</font><font color="#993399">1</font><font color="#990000">))?</font> i<font color="#990000">+</font><font color="#993399">1</font> <font color="#990000">:</font> <font color="#993399">0</font><font color="#990000">;</font>
<b><font color="#0000FF">if</font></b> <font color="#990000">(</font> isPow2 <font color="#990000">)</font>
cout <font color="#990000"><<</font> endl <font color="#990000"><<</font> <font color="#FF0000">"</font><font color="#CC33CC">\t</font><font color="#FF0000">"</font><font color="#990000">;</font>
<font color="#FF0000">}</font>
cout <font color="#990000"><<</font> endl<font color="#990000">;</font>
<font color="#FF0000">}</font>
</tt></pre>
</body>
</html>