-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathPairHeap.java
More file actions
252 lines (223 loc) · 8.71 KB
/
Copy pathPairHeap.java
File metadata and controls
252 lines (223 loc) · 8.71 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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
package DataStructures.Heap;
import DataStructures.Comparable;
import DataStructures.MyInteger;
// PairHeap class
//
// CONSTRUCTION: with no initializer
//
// ******************PUBLIC OPERATIONS*********************
// PairNode insert( x ) --> Insert x, return position
// Comparable deleteMin( )--> Return and remove smallest item
// Comparable findMin( ) --> Return smallest item
// boolean isEmpty( ) --> Return true if empty; else false
// void makeEmpty( ) --> Remove all items
// void decreaseKey( PairNode p, newVal )
// --> Decrease value in node p
/**
* Implements a pairing heap.
* Supports a decreaseKey operation.
* Note that all "matching" is based on the compareTo method.
* @author Mark Allen Weiss
* @see PairNode
*/
public class PairHeap
{
/**
* Construct the pairing heap.
*/
public PairHeap( )
{
root = null;
}
/**
* Insert into the priority queue, and return a PairNode
* that can be used by decreaseKey.
* Duplicates are allowed.
* @param x the item to insert.
* @return the node containing the newly inserted item.
*/
public PairNode insert( Comparable x )
{
PairNode newNode = new PairNode( x );
if( root == null )
root = newNode;
else
root = compareAndLink( root, newNode );
return newNode;
}
/**
* Find the smallest item in the priority queue.
* @return the smallest item, or null if empty.
*/
public Comparable findMin( )
{
if( isEmpty( ) )
return null;
return root.element;
}
/**
* Remove the smallest item from the priority queue.
* @return the smallest item, or null if empty.
*/
public Comparable deleteMin( )
{
if( isEmpty( ) )
return null;
Comparable x = findMin( );
if( root.leftChild == null )
root = null;
else
root = combineSiblings( root.leftChild );
return x;
}
/**
* Change the value of the item stored in the pairing heap.
* Does nothing if newVal is larger than the currently stored value.
* @param p any node returned by addItem.
* @param newVal the new value, which must be smaller
* than the currently stored value.
*/
public void decreaseKey( PairNode p, Comparable newVal )
{
if( p.element.compareTo( newVal ) < 0 )
return; // newVal cannot be bigger
p.element = newVal;
if( p != root )
{
if( p.nextSibling != null )
p.nextSibling.prev = p.prev;
if( p.prev.leftChild == p )
p.prev.leftChild = p.nextSibling;
else
p.prev.nextSibling = p.nextSibling;
p.nextSibling = null;
root = compareAndLink( root, p );
}
}
/**
* Test if the priority queue is logically empty.
* @return true if empty, false otherwise.
*/
public boolean isEmpty( )
{
return root == null;
}
/**
* Make the priority queue logically empty.
*/
public void makeEmpty( )
{
root = null;
}
private PairNode root;
/**
* Internal method that is the basic operation to maintain order.
* Links first and second together to satisfy heap order.
* @param first root of tree 1, which may not be null.
* first.nextSibling MUST be null on entry.
* @param second root of tree 2, which may be null.
* @return result of the tree merge.
*/
private PairNode compareAndLink( PairNode first, PairNode second )
{
if( second == null )
return first;
if( second.element.compareTo( first.element ) < 0 )
{
// Attach first as leftmost child of second
second.prev = first.prev;
first.prev = second;
first.nextSibling = second.leftChild;
if( first.nextSibling != null )
first.nextSibling.prev = first;
second.leftChild = first;
return second;
}
else
{
// Attach second as leftmost child of first
second.prev = first;
first.nextSibling = second.nextSibling;
if( first.nextSibling != null )
first.nextSibling.prev = first;
second.nextSibling = first.leftChild;
if( second.nextSibling != null )
second.nextSibling.prev = second;
first.leftChild = second;
return first;
}
}
private PairNode [ ] doubleIfFull( PairNode [ ] array, int index )
{
if( index == array.length )
{
PairNode [ ] oldArray = array;
array = new PairNode[ index * 2 ];
for( int i = 0; i < index; i++ )
array[ i ] = oldArray[ i ];
}
return array;
}
// The tree array for combineSiblings
private PairNode [ ] treeArray = new PairNode[ 5 ];
/**
* Internal method that implements two-pass merging.
* @param firstSibling the root of the conglomerate;
* assumed not null.
*/
private PairNode combineSiblings( PairNode firstSibling )
{
if( firstSibling.nextSibling == null )
return firstSibling;
// Store the subtrees in an array
int numSiblings = 0;
for( ; firstSibling != null; numSiblings++ )
{
treeArray = doubleIfFull( treeArray, numSiblings );
treeArray[ numSiblings ] = firstSibling;
firstSibling.prev.nextSibling = null; // break links
firstSibling = firstSibling.nextSibling;
}
treeArray = doubleIfFull( treeArray, numSiblings );
treeArray[ numSiblings ] = null;
// Combine subtrees two at a time, going left to right
int i = 0;
for( ; i + 1 < numSiblings; i += 2 )
treeArray[ i ] = compareAndLink( treeArray[ i ], treeArray[ i + 1 ] );
int j = i - 2;
// j has the result of last compareAndLink.
// If an odd number of trees, get the last one.
if( j == numSiblings - 3 )
treeArray[ j ] = compareAndLink( treeArray[ j ], treeArray[ j + 2 ] );
// Now go right to left, merging last tree with
// next to last. The result becomes the new last.
for( ; j >= 2; j -= 2 )
treeArray[ j - 2 ] = compareAndLink( treeArray[ j - 2 ], treeArray[ j ] );
return treeArray[ 0 ];
}
// Test program
public static void main( String [ ] args )
{
PairHeap h = new PairHeap( );
int numItems = 10000;
int i = 37;
int j;
System.out.println( "Checking; no bad output is good" );
for( i = 37; i != 0; i = ( i + 37 ) % numItems )
h.insert( new MyInteger( i ) );
for( i = 1; i < numItems; i++ )
if( ((MyInteger)( h.deleteMin( ) )).intValue( ) != i )
System.out.println( "Oops! " + i );
PairNode [ ] p = new PairNode[ numItems ];
for( i = 0, j = numItems / 2; i < numItems; i++, j =(j+71)%numItems )
p[ j ] = h.insert( new MyInteger( j + numItems ) );
for( i = 0, j = numItems / 2; i < numItems; i++, j =(j+53)%numItems )
h.decreaseKey( p[ j ], new MyInteger(
((MyInteger)p[ j ].element).intValue( ) - numItems ) );
i = -1;
while( !h.isEmpty( ) )
if( ((MyInteger)( h.deleteMin( ) )).intValue( ) != ++i )
System.out.println( "Oops! " + i + " " );
System.out.println( "Check completed" );
}
}