1 | #region License
|
---|
2 | /*
|
---|
3 | The MIT License (MIT)
|
---|
4 |
|
---|
5 | Copyright (c) 2013 Daniel "BlueRaja" Pflughoeft
|
---|
6 |
|
---|
7 | Permission is hereby granted, free of charge, to any person obtaining a copy
|
---|
8 | of this software and associated documentation files (the "Software"), to deal
|
---|
9 | in the Software without restriction, including without limitation the rights
|
---|
10 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
---|
11 | copies of the Software, and to permit persons to whom the Software is
|
---|
12 | furnished to do so, subject to the following conditions:
|
---|
13 |
|
---|
14 | The above copyright notice and this permission notice shall be included in
|
---|
15 | all copies or substantial portions of the Software.
|
---|
16 |
|
---|
17 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
---|
18 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
---|
19 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
---|
20 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
---|
21 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
---|
22 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
---|
23 | THE SOFTWARE.
|
---|
24 | */
|
---|
25 | #endregion
|
---|
26 |
|
---|
27 | using System;
|
---|
28 | using System.Collections;
|
---|
29 | using System.Collections.Generic;
|
---|
30 | using System.Linq;
|
---|
31 |
|
---|
32 | namespace SimSharp {
|
---|
33 | /// <summary>
|
---|
34 | /// A simplified priority queue implementation. Is stable, auto-resizes, and thread-safe, at the cost of being slightly slower than
|
---|
35 | /// FastPriorityQueue
|
---|
36 | /// Methods tagged as O(1) or O(log n) are assuming there are no duplicates. Duplicates may increase the algorithmic complexity.
|
---|
37 | /// </summary>
|
---|
38 | /// <remarks>
|
---|
39 | /// Original sources from https://github.com/BlueRaja/High-Speed-Priority-Queue-for-C-Sharp
|
---|
40 | /// </remarks>
|
---|
41 | /// <typeparam name="TItem">The type to enqueue</typeparam>
|
---|
42 | /// <typeparam name="TPriority">The priority-type to use for nodes. Must extend IComparable<TPriority></typeparam>
|
---|
43 | public class SimplePriorityQueue<TItem, TPriority> : IEnumerable<TItem>
|
---|
44 | where TPriority : IComparable<TPriority> {
|
---|
45 | private class SimpleNode : GenericPriorityQueueNode<TPriority> {
|
---|
46 | public TItem Data { get; private set; }
|
---|
47 |
|
---|
48 | public SimpleNode(TItem data) {
|
---|
49 | Data = data;
|
---|
50 | }
|
---|
51 | }
|
---|
52 |
|
---|
53 | private const int INITIAL_QUEUE_SIZE = 64;
|
---|
54 | private readonly GenericPriorityQueue<SimpleNode, TPriority> _queue;
|
---|
55 | private readonly Dictionary<TItem, IList<SimpleNode>> _itemToNodesCache;
|
---|
56 | private readonly IList<SimpleNode> _nullNodesCache;
|
---|
57 |
|
---|
58 | /// <summary>
|
---|
59 | /// Instantiate a new Priority Queue
|
---|
60 | /// </summary>
|
---|
61 | public SimplePriorityQueue() : this(Comparer<TPriority>.Default) { }
|
---|
62 |
|
---|
63 | /// <summary>
|
---|
64 | /// Instantiate a new Priority Queue
|
---|
65 | /// </summary>
|
---|
66 | /// <param name="comparer">The comparer used to compare TPriority values. Defaults to Comparer<TPriority>.default</param>
|
---|
67 | public SimplePriorityQueue(IComparer<TPriority> comparer) : this(comparer.Compare) { }
|
---|
68 |
|
---|
69 | /// <summary>
|
---|
70 | /// Instantiate a new Priority Queue
|
---|
71 | /// </summary>
|
---|
72 | /// <param name="comparer">The comparison function to use to compare TPriority values</param>
|
---|
73 | public SimplePriorityQueue(Comparison<TPriority> comparer) {
|
---|
74 | _queue = new GenericPriorityQueue<SimpleNode, TPriority>(INITIAL_QUEUE_SIZE, comparer);
|
---|
75 | _itemToNodesCache = new Dictionary<TItem, IList<SimpleNode>>();
|
---|
76 | _nullNodesCache = new List<SimpleNode>();
|
---|
77 | }
|
---|
78 |
|
---|
79 | /// <summary>
|
---|
80 | /// Given an item of type T, returns the exist SimpleNode in the queue
|
---|
81 | /// </summary>
|
---|
82 | private SimpleNode GetExistingNode(TItem item) {
|
---|
83 | if (item == null) {
|
---|
84 | return _nullNodesCache.Count > 0 ? _nullNodesCache[0] : null;
|
---|
85 | }
|
---|
86 |
|
---|
87 | IList<SimpleNode> nodes;
|
---|
88 | if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
89 | return null;
|
---|
90 | }
|
---|
91 | return nodes[0];
|
---|
92 | }
|
---|
93 |
|
---|
94 | /// <summary>
|
---|
95 | /// Adds an item to the Node-cache to allow for many methods to be O(1) or O(log n)
|
---|
96 | /// </summary>
|
---|
97 | private void AddToNodeCache(SimpleNode node) {
|
---|
98 | if (node.Data == null) {
|
---|
99 | _nullNodesCache.Add(node);
|
---|
100 | return;
|
---|
101 | }
|
---|
102 |
|
---|
103 | IList<SimpleNode> nodes;
|
---|
104 | if (!_itemToNodesCache.TryGetValue(node.Data, out nodes)) {
|
---|
105 | nodes = new List<SimpleNode>();
|
---|
106 | _itemToNodesCache[node.Data] = nodes;
|
---|
107 | }
|
---|
108 | nodes.Add(node);
|
---|
109 | }
|
---|
110 |
|
---|
111 | /// <summary>
|
---|
112 | /// Removes an item to the Node-cache to allow for many methods to be O(1) or O(log n) (assuming no duplicates)
|
---|
113 | /// </summary>
|
---|
114 | private void RemoveFromNodeCache(SimpleNode node) {
|
---|
115 | if (node.Data == null) {
|
---|
116 | _nullNodesCache.Remove(node);
|
---|
117 | return;
|
---|
118 | }
|
---|
119 |
|
---|
120 | IList<SimpleNode> nodes;
|
---|
121 | if (!_itemToNodesCache.TryGetValue(node.Data, out nodes)) {
|
---|
122 | return;
|
---|
123 | }
|
---|
124 | nodes.Remove(node);
|
---|
125 | if (nodes.Count == 0) {
|
---|
126 | _itemToNodesCache.Remove(node.Data);
|
---|
127 | }
|
---|
128 | }
|
---|
129 |
|
---|
130 | /// <summary>
|
---|
131 | /// Returns the number of nodes in the queue.
|
---|
132 | /// O(1)
|
---|
133 | /// </summary>
|
---|
134 | public int Count {
|
---|
135 | get {
|
---|
136 | return _queue.Count;
|
---|
137 | }
|
---|
138 | }
|
---|
139 |
|
---|
140 |
|
---|
141 | /// <summary>
|
---|
142 | /// Returns the head of the queue, without removing it (use Dequeue() for that).
|
---|
143 | /// Throws an exception when the queue is empty.
|
---|
144 | /// O(1)
|
---|
145 | /// </summary>
|
---|
146 | public TItem First {
|
---|
147 | get {
|
---|
148 | if (_queue.Count <= 0) {
|
---|
149 | throw new InvalidOperationException("Cannot call .First on an empty queue");
|
---|
150 | }
|
---|
151 |
|
---|
152 | return _queue.First.Data;
|
---|
153 | }
|
---|
154 | }
|
---|
155 |
|
---|
156 | /// <summary>
|
---|
157 | /// Returns the top priority of the queue.
|
---|
158 | /// Throws an exception when the queue is empty.
|
---|
159 | /// O(1)
|
---|
160 | /// </summary>
|
---|
161 | public TPriority Peek {
|
---|
162 | get {
|
---|
163 | lock (_queue) {
|
---|
164 | if (_queue.Count <= 0) {
|
---|
165 | throw new InvalidOperationException("Cannot call .Peek on an empty queue");
|
---|
166 | }
|
---|
167 | return _queue.First.Priority;
|
---|
168 | }
|
---|
169 | }
|
---|
170 | }
|
---|
171 |
|
---|
172 | /// <summary>
|
---|
173 | /// Removes every node from the queue.
|
---|
174 | /// O(n)
|
---|
175 | /// </summary>
|
---|
176 | public void Clear() {
|
---|
177 | _queue.Clear();
|
---|
178 | _itemToNodesCache.Clear();
|
---|
179 | _nullNodesCache.Clear();
|
---|
180 | }
|
---|
181 |
|
---|
182 | /// <summary>
|
---|
183 | /// Returns whether the given item is in the queue.
|
---|
184 | /// O(1)
|
---|
185 | /// </summary>
|
---|
186 | public bool Contains(TItem item) {
|
---|
187 | if (item == null) {
|
---|
188 | return _nullNodesCache.Count > 0;
|
---|
189 | }
|
---|
190 | return _itemToNodesCache.ContainsKey(item);
|
---|
191 | }
|
---|
192 |
|
---|
193 | /// <summary>
|
---|
194 | /// Removes the head of the queue (node with minimum priority; ties are broken by order of insertion), and returns it.
|
---|
195 | /// If queue is empty, throws an exception
|
---|
196 | /// O(log n)
|
---|
197 | /// </summary>
|
---|
198 | public TItem Dequeue() {
|
---|
199 | if (_queue.Count <= 0) {
|
---|
200 | throw new InvalidOperationException("Cannot call Dequeue() on an empty queue");
|
---|
201 | }
|
---|
202 |
|
---|
203 | SimpleNode node = _queue.Dequeue();
|
---|
204 | RemoveFromNodeCache(node);
|
---|
205 | return node.Data;
|
---|
206 | }
|
---|
207 |
|
---|
208 | /// <summary>
|
---|
209 | /// Enqueue the item with the given priority, without calling AddToNodeCache(node)
|
---|
210 | /// </summary>
|
---|
211 | /// <param name="item"></param>
|
---|
212 | /// <param name="priority"></param>
|
---|
213 | /// <returns></returns>
|
---|
214 | private SimpleNode EnqueueNoCache(TItem item, TPriority priority) {
|
---|
215 | SimpleNode node = new SimpleNode(item);
|
---|
216 | if (_queue.Count == _queue.MaxSize) {
|
---|
217 | _queue.Resize(_queue.MaxSize * 2 + 1);
|
---|
218 | }
|
---|
219 | _queue.Enqueue(node, priority);
|
---|
220 | return node;
|
---|
221 | }
|
---|
222 |
|
---|
223 | /// <summary>
|
---|
224 | /// Enqueue a node to the priority queue. Lower values are placed in front. Ties are broken by first-in-first-out.
|
---|
225 | /// This queue automatically resizes itself, so there's no concern of the queue becoming 'full'.
|
---|
226 | /// Duplicates and null-values are allowed.
|
---|
227 | /// O(log n)
|
---|
228 | /// </summary>
|
---|
229 | public void Enqueue(TItem item, TPriority priority) {
|
---|
230 | IList<SimpleNode> nodes;
|
---|
231 | if (item == null) {
|
---|
232 | nodes = _nullNodesCache;
|
---|
233 | } else if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
234 | nodes = new List<SimpleNode>();
|
---|
235 | _itemToNodesCache[item] = nodes;
|
---|
236 | }
|
---|
237 | SimpleNode node = EnqueueNoCache(item, priority);
|
---|
238 | nodes.Add(node);
|
---|
239 | }
|
---|
240 |
|
---|
241 | /// <summary>
|
---|
242 | /// Enqueue a node to the priority queue if it doesn't already exist. Lower values are placed in front. Ties are broken by first-in-first-out.
|
---|
243 | /// This queue automatically resizes itself, so there's no concern of the queue becoming 'full'. Null values are allowed.
|
---|
244 | /// Returns true if the node was successfully enqueued; false if it already exists.
|
---|
245 | /// O(log n)
|
---|
246 | /// </summary>
|
---|
247 | public bool EnqueueWithoutDuplicates(TItem item, TPriority priority) {
|
---|
248 | IList<SimpleNode> nodes;
|
---|
249 | if (item == null) {
|
---|
250 | if (_nullNodesCache.Count > 0) {
|
---|
251 | return false;
|
---|
252 | }
|
---|
253 | nodes = _nullNodesCache;
|
---|
254 | } else if (_itemToNodesCache.ContainsKey(item)) {
|
---|
255 | return false;
|
---|
256 | } else {
|
---|
257 | nodes = new List<SimpleNode>();
|
---|
258 | _itemToNodesCache[item] = nodes;
|
---|
259 | }
|
---|
260 | SimpleNode node = EnqueueNoCache(item, priority);
|
---|
261 | nodes.Add(node);
|
---|
262 | return true;
|
---|
263 | }
|
---|
264 |
|
---|
265 | /// <summary>
|
---|
266 | /// Removes an item from the queue. The item does not need to be the head of the queue.
|
---|
267 | /// If the item is not in the queue, an exception is thrown. If unsure, check Contains() first.
|
---|
268 | /// If multiple copies of the item are enqueued, only the first one is removed.
|
---|
269 | /// O(log n)
|
---|
270 | /// </summary>
|
---|
271 | public void Remove(TItem item) {
|
---|
272 | SimpleNode removeMe;
|
---|
273 | IList<SimpleNode> nodes;
|
---|
274 | if (item == null) {
|
---|
275 | if (_nullNodesCache.Count == 0) {
|
---|
276 | throw new InvalidOperationException("Cannot call Remove() on a node which is not enqueued: " + item);
|
---|
277 | }
|
---|
278 | removeMe = _nullNodesCache[0];
|
---|
279 | nodes = _nullNodesCache;
|
---|
280 | } else {
|
---|
281 | if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
282 | throw new InvalidOperationException("Cannot call Remove() on a node which is not enqueued: " + item);
|
---|
283 | }
|
---|
284 | removeMe = nodes[0];
|
---|
285 | if (nodes.Count == 1) {
|
---|
286 | _itemToNodesCache.Remove(item);
|
---|
287 | }
|
---|
288 | }
|
---|
289 | _queue.Remove(removeMe);
|
---|
290 | nodes.Remove(removeMe);
|
---|
291 | }
|
---|
292 |
|
---|
293 | /// <summary>
|
---|
294 | /// Call this method to change the priority of an item.
|
---|
295 | /// Calling this method on a item not in the queue will throw an exception.
|
---|
296 | /// If the item is enqueued multiple times, only the first one will be updated.
|
---|
297 | /// (If your requirements are complex enough that you need to enqueue the same item multiple times <i>and</i> be able
|
---|
298 | /// to update all of them, please wrap your items in a wrapper class so they can be distinguished).
|
---|
299 | /// O(log n)
|
---|
300 | /// </summary>
|
---|
301 | public void UpdatePriority(TItem item, TPriority priority) {
|
---|
302 | SimpleNode updateMe = GetExistingNode(item);
|
---|
303 | if (updateMe == null) {
|
---|
304 | throw new InvalidOperationException("Cannot call UpdatePriority() on a node which is not enqueued: " + item);
|
---|
305 | }
|
---|
306 | _queue.UpdatePriority(updateMe, priority);
|
---|
307 | }
|
---|
308 |
|
---|
309 | /// <summary>
|
---|
310 | /// Returns the priority of the given item.
|
---|
311 | /// Calling this method on a item not in the queue will throw an exception.
|
---|
312 | /// If the item is enqueued multiple times, only the priority of the first will be returned.
|
---|
313 | /// (If your requirements are complex enough that you need to enqueue the same item multiple times <i>and</i> be able
|
---|
314 | /// to query all their priorities, please wrap your items in a wrapper class so they can be distinguished).
|
---|
315 | /// O(1)
|
---|
316 | /// </summary>
|
---|
317 | public TPriority GetPriority(TItem item) {
|
---|
318 | SimpleNode findMe = GetExistingNode(item);
|
---|
319 | if (findMe == null) {
|
---|
320 | throw new InvalidOperationException("Cannot call GetPriority() on a node which is not enqueued: " + item);
|
---|
321 | }
|
---|
322 | return findMe.Priority;
|
---|
323 | }
|
---|
324 |
|
---|
325 | #region Try* methods for multithreading
|
---|
326 | /// Get the head of the queue, without removing it (use TryDequeue() for that).
|
---|
327 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and First
|
---|
328 | /// Returns true if successful, false otherwise
|
---|
329 | /// O(1)
|
---|
330 | public bool TryFirst(out TItem first) {
|
---|
331 | if (_queue.Count <= 0) {
|
---|
332 | first = default(TItem);
|
---|
333 | return false;
|
---|
334 | }
|
---|
335 |
|
---|
336 | first = _queue.First.Data;
|
---|
337 | return true;
|
---|
338 | }
|
---|
339 |
|
---|
340 | /// <summary>
|
---|
341 | /// Removes the head of the queue (node with minimum priority; ties are broken by order of insertion), and sets it to first.
|
---|
342 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and Dequeue()
|
---|
343 | /// Returns true if successful; false if queue was empty
|
---|
344 | /// O(log n)
|
---|
345 | /// </summary>
|
---|
346 | public bool TryDequeue(out TItem first) {
|
---|
347 | if (_queue.Count <= 0) {
|
---|
348 | first = default(TItem);
|
---|
349 | return false;
|
---|
350 | }
|
---|
351 |
|
---|
352 | SimpleNode node = _queue.Dequeue();
|
---|
353 | first = node.Data;
|
---|
354 | RemoveFromNodeCache(node);
|
---|
355 | return true;
|
---|
356 | }
|
---|
357 |
|
---|
358 | /// <summary>
|
---|
359 | /// Attempts to remove an item from the queue. The item does not need to be the head of the queue.
|
---|
360 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and Remove()
|
---|
361 | /// Returns true if the item was successfully removed, false if it wasn't in the queue.
|
---|
362 | /// If multiple copies of the item are enqueued, only the first one is removed.
|
---|
363 | /// O(log n)
|
---|
364 | /// </summary>
|
---|
365 | public bool TryRemove(TItem item) {
|
---|
366 | SimpleNode removeMe;
|
---|
367 | IList<SimpleNode> nodes;
|
---|
368 | if (item == null) {
|
---|
369 | if (_nullNodesCache.Count == 0) {
|
---|
370 | return false;
|
---|
371 | }
|
---|
372 | removeMe = _nullNodesCache[0];
|
---|
373 | nodes = _nullNodesCache;
|
---|
374 | } else {
|
---|
375 | if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
376 | return false;
|
---|
377 | }
|
---|
378 | removeMe = nodes[0];
|
---|
379 | if (nodes.Count == 1) {
|
---|
380 | _itemToNodesCache.Remove(item);
|
---|
381 | }
|
---|
382 | }
|
---|
383 | _queue.Remove(removeMe);
|
---|
384 | nodes.Remove(removeMe);
|
---|
385 | return true;
|
---|
386 | }
|
---|
387 |
|
---|
388 | /// <summary>
|
---|
389 | /// Call this method to change the priority of an item.
|
---|
390 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and UpdatePriority()
|
---|
391 | /// If the item is enqueued multiple times, only the first one will be updated.
|
---|
392 | /// (If your requirements are complex enough that you need to enqueue the same item multiple times <i>and</i> be able
|
---|
393 | /// to update all of them, please wrap your items in a wrapper class so they can be distinguished).
|
---|
394 | /// Returns true if the item priority was updated, false otherwise.
|
---|
395 | /// O(log n)
|
---|
396 | /// </summary>
|
---|
397 | public bool TryUpdatePriority(TItem item, TPriority priority) {
|
---|
398 | SimpleNode updateMe = GetExistingNode(item);
|
---|
399 | if (updateMe == null) {
|
---|
400 | return false;
|
---|
401 | }
|
---|
402 | _queue.UpdatePriority(updateMe, priority);
|
---|
403 | return true;
|
---|
404 | }
|
---|
405 |
|
---|
406 | /// <summary>
|
---|
407 | /// Attempt to get the priority of the given item.
|
---|
408 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and GetPriority()
|
---|
409 | /// If the item is enqueued multiple times, only the priority of the first will be returned.
|
---|
410 | /// (If your requirements are complex enough that you need to enqueue the same item multiple times <i>and</i> be able
|
---|
411 | /// to query all their priorities, please wrap your items in a wrapper class so they can be distinguished).
|
---|
412 | /// Returns true if the item was found in the queue, false otherwise
|
---|
413 | /// O(1)
|
---|
414 | /// </summary>
|
---|
415 | public bool TryGetPriority(TItem item, out TPriority priority) {
|
---|
416 | SimpleNode findMe = GetExistingNode(item);
|
---|
417 | if (findMe == null) {
|
---|
418 | priority = default(TPriority);
|
---|
419 | return false;
|
---|
420 | }
|
---|
421 | priority = findMe.Priority;
|
---|
422 | return true;
|
---|
423 | }
|
---|
424 | #endregion
|
---|
425 |
|
---|
426 | public IEnumerator<TItem> GetEnumerator() {
|
---|
427 | return _queue.Select(x => x.Data).GetEnumerator();
|
---|
428 | }
|
---|
429 |
|
---|
430 | IEnumerator IEnumerable.GetEnumerator() {
|
---|
431 | return GetEnumerator();
|
---|
432 | }
|
---|
433 |
|
---|
434 | public bool IsValidQueue() {
|
---|
435 | // Check all items in cache are in the queue
|
---|
436 | foreach (IList<SimpleNode> nodes in _itemToNodesCache.Values) {
|
---|
437 | foreach (SimpleNode node in nodes) {
|
---|
438 | if (!_queue.Contains(node)) {
|
---|
439 | return false;
|
---|
440 | }
|
---|
441 | }
|
---|
442 | }
|
---|
443 |
|
---|
444 | // Check all items in queue are in cache
|
---|
445 | foreach (SimpleNode node in _queue) {
|
---|
446 | if (GetExistingNode(node.Data) == null) {
|
---|
447 | return false;
|
---|
448 | }
|
---|
449 | }
|
---|
450 |
|
---|
451 | // Check queue structure itself
|
---|
452 | return _queue.IsValidQueue();
|
---|
453 | }
|
---|
454 | }
|
---|
455 |
|
---|
456 | /// <summary>
|
---|
457 | /// A simplified priority queue implementation. Is stable, auto-resizes, and thread-safe, at the cost of being slightly slower than
|
---|
458 | /// FastPriorityQueue
|
---|
459 | /// Methods tagged as O(1) or O(log n) are assuming there are no duplicates. Duplicates may increase the algorithmic complexity.
|
---|
460 | /// </summary>
|
---|
461 | /// <typeparam name="TItem">The type to enqueue</typeparam>
|
---|
462 | public class SimplePriorityQueue<TItem> : IEnumerable<TItem> where TItem : IComparable<TItem> {
|
---|
463 | private class SimpleNode : GenericPriorityQueueNode, IComparable<SimpleNode>, IComparable {
|
---|
464 | public TItem Data { get; private set; }
|
---|
465 | private Comparison<TItem> _comparer;
|
---|
466 |
|
---|
467 | public SimpleNode(TItem data, Comparison<TItem> comparer) {
|
---|
468 | Data = data;
|
---|
469 | _comparer = comparer;
|
---|
470 | }
|
---|
471 |
|
---|
472 | public int CompareTo(SimpleNode other) {
|
---|
473 | if (other == null) return 1;
|
---|
474 | return _comparer(Data, other.Data);
|
---|
475 | }
|
---|
476 |
|
---|
477 | public int CompareTo(object obj) {
|
---|
478 | if (obj is SimpleNode other) return CompareTo(other);
|
---|
479 | if (obj == null) return 1;
|
---|
480 | throw new ArgumentException("can only compare to objects of type SimpleNode.");
|
---|
481 | }
|
---|
482 | }
|
---|
483 |
|
---|
484 | private const int INITIAL_QUEUE_SIZE = 64;
|
---|
485 | private readonly GenericPriorityQueue<SimpleNode> _queue;
|
---|
486 | private readonly Dictionary<TItem, IList<SimpleNode>> _itemToNodesCache;
|
---|
487 | private readonly IList<SimpleNode> _nullNodesCache;
|
---|
488 | private readonly Comparison<TItem> _comparer;
|
---|
489 |
|
---|
490 | /// <summary>
|
---|
491 | /// Instantiate a new Priority Queue
|
---|
492 | /// </summary>
|
---|
493 | public SimplePriorityQueue() : this(Comparer<TItem>.Default) { }
|
---|
494 |
|
---|
495 | /// <summary>
|
---|
496 | /// Instantiate a new Priority Queue
|
---|
497 | /// </summary>
|
---|
498 | /// <param name="comparer">The comparer used to compare TPriority values. Defaults to Comparer<TPriority>.default</param>
|
---|
499 | public SimplePriorityQueue(IComparer<TItem> comparer) : this(comparer.Compare) { }
|
---|
500 |
|
---|
501 | /// <summary>
|
---|
502 | /// Instantiate a new Priority Queue
|
---|
503 | /// </summary>
|
---|
504 | /// <param name="comparer">The comparison function to use to compare TPriority values</param>
|
---|
505 | public SimplePriorityQueue(Comparison<TItem> comparer) {
|
---|
506 | _comparer = comparer;
|
---|
507 | // the inner priority queue will always use the default comparer which in turn makes use of the specified comparer
|
---|
508 | _queue = new GenericPriorityQueue<SimpleNode>(INITIAL_QUEUE_SIZE, Comparer<SimpleNode>.Default);
|
---|
509 | _itemToNodesCache = new Dictionary<TItem, IList<SimpleNode>>();
|
---|
510 | _nullNodesCache = new List<SimpleNode>();
|
---|
511 | }
|
---|
512 |
|
---|
513 | /// <summary>
|
---|
514 | /// Given an item of type T, returns the exist SimpleNode in the queue
|
---|
515 | /// </summary>
|
---|
516 | private SimpleNode GetExistingNode(TItem item) {
|
---|
517 | if (item == null) {
|
---|
518 | return _nullNodesCache.Count > 0 ? _nullNodesCache[0] : null;
|
---|
519 | }
|
---|
520 |
|
---|
521 | IList<SimpleNode> nodes;
|
---|
522 | if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
523 | return null;
|
---|
524 | }
|
---|
525 | return nodes[0];
|
---|
526 | }
|
---|
527 |
|
---|
528 | /// <summary>
|
---|
529 | /// Adds an item to the Node-cache to allow for many methods to be O(1) or O(log n)
|
---|
530 | /// </summary>
|
---|
531 | private void AddToNodeCache(SimpleNode node) {
|
---|
532 | if (node.Data == null) {
|
---|
533 | _nullNodesCache.Add(node);
|
---|
534 | return;
|
---|
535 | }
|
---|
536 |
|
---|
537 | IList<SimpleNode> nodes;
|
---|
538 | if (!_itemToNodesCache.TryGetValue(node.Data, out nodes)) {
|
---|
539 | nodes = new List<SimpleNode>();
|
---|
540 | _itemToNodesCache[node.Data] = nodes;
|
---|
541 | }
|
---|
542 | nodes.Add(node);
|
---|
543 | }
|
---|
544 |
|
---|
545 | /// <summary>
|
---|
546 | /// Removes an item to the Node-cache to allow for many methods to be O(1) or O(log n) (assuming no duplicates)
|
---|
547 | /// </summary>
|
---|
548 | private void RemoveFromNodeCache(SimpleNode node) {
|
---|
549 | if (node.Data == null) {
|
---|
550 | _nullNodesCache.Remove(node);
|
---|
551 | return;
|
---|
552 | }
|
---|
553 |
|
---|
554 | IList<SimpleNode> nodes;
|
---|
555 | if (!_itemToNodesCache.TryGetValue(node.Data, out nodes)) {
|
---|
556 | return;
|
---|
557 | }
|
---|
558 | nodes.Remove(node);
|
---|
559 | if (nodes.Count == 0) {
|
---|
560 | _itemToNodesCache.Remove(node.Data);
|
---|
561 | }
|
---|
562 | }
|
---|
563 |
|
---|
564 | /// <summary>
|
---|
565 | /// Returns the number of nodes in the queue.
|
---|
566 | /// O(1)
|
---|
567 | /// </summary>
|
---|
568 | public int Count {
|
---|
569 | get {
|
---|
570 | return _queue.Count;
|
---|
571 | }
|
---|
572 | }
|
---|
573 |
|
---|
574 |
|
---|
575 | /// <summary>
|
---|
576 | /// Returns the head of the queue, without removing it (use Dequeue() for that).
|
---|
577 | /// Throws an exception when the queue is empty.
|
---|
578 | /// O(1)
|
---|
579 | /// </summary>
|
---|
580 | public TItem First {
|
---|
581 | get {
|
---|
582 | if (_queue.Count <= 0) {
|
---|
583 | throw new InvalidOperationException("Cannot call .First on an empty queue");
|
---|
584 | }
|
---|
585 |
|
---|
586 | return _queue.First.Data;
|
---|
587 | }
|
---|
588 | }
|
---|
589 |
|
---|
590 | /// <summary>
|
---|
591 | /// Removes every node from the queue.
|
---|
592 | /// O(n)
|
---|
593 | /// </summary>
|
---|
594 | public void Clear() {
|
---|
595 | _queue.Clear();
|
---|
596 | _itemToNodesCache.Clear();
|
---|
597 | _nullNodesCache.Clear();
|
---|
598 | }
|
---|
599 |
|
---|
600 | /// <summary>
|
---|
601 | /// Returns whether the given item is in the queue.
|
---|
602 | /// O(1)
|
---|
603 | /// </summary>
|
---|
604 | public bool Contains(TItem item) {
|
---|
605 | if (item == null) {
|
---|
606 | return _nullNodesCache.Count > 0;
|
---|
607 | }
|
---|
608 | return _itemToNodesCache.ContainsKey(item);
|
---|
609 | }
|
---|
610 |
|
---|
611 | /// <summary>
|
---|
612 | /// Removes the head of the queue (node with minimum priority; ties are broken by order of insertion), and returns it.
|
---|
613 | /// If queue is empty, throws an exception
|
---|
614 | /// O(log n)
|
---|
615 | /// </summary>
|
---|
616 | public TItem Dequeue() {
|
---|
617 | if (_queue.Count <= 0) {
|
---|
618 | throw new InvalidOperationException("Cannot call Dequeue() on an empty queue");
|
---|
619 | }
|
---|
620 |
|
---|
621 | SimpleNode node = _queue.Dequeue();
|
---|
622 | RemoveFromNodeCache(node);
|
---|
623 | return node.Data;
|
---|
624 | }
|
---|
625 |
|
---|
626 | /// <summary>
|
---|
627 | /// Enqueue the item with the given priority, without calling AddToNodeCache(node)
|
---|
628 | /// </summary>
|
---|
629 | /// <param name="item"></param>
|
---|
630 | /// <returns></returns>
|
---|
631 | private SimpleNode EnqueueNoCache(TItem item) {
|
---|
632 | SimpleNode node = new SimpleNode(item, _comparer);
|
---|
633 | if (_queue.Count == _queue.MaxSize) {
|
---|
634 | _queue.Resize(_queue.MaxSize * 2 + 1);
|
---|
635 | }
|
---|
636 | _queue.Enqueue(node);
|
---|
637 | return node;
|
---|
638 | }
|
---|
639 |
|
---|
640 | /// <summary>
|
---|
641 | /// Enqueue a node to the priority queue. Lower values are placed in front. Ties are broken by first-in-first-out.
|
---|
642 | /// This queue automatically resizes itself, so there's no concern of the queue becoming 'full'.
|
---|
643 | /// Duplicates and null-values are allowed.
|
---|
644 | /// O(log n)
|
---|
645 | /// </summary>
|
---|
646 | public void Enqueue(TItem item) {
|
---|
647 | IList<SimpleNode> nodes;
|
---|
648 | if (item == null) {
|
---|
649 | nodes = _nullNodesCache;
|
---|
650 | } else if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
651 | nodes = new List<SimpleNode>();
|
---|
652 | _itemToNodesCache[item] = nodes;
|
---|
653 | }
|
---|
654 | SimpleNode node = EnqueueNoCache(item);
|
---|
655 | nodes.Add(node);
|
---|
656 | }
|
---|
657 |
|
---|
658 | /// <summary>
|
---|
659 | /// Enqueue a node to the priority queue if it doesn't already exist. Lower values are placed in front. Ties are broken by first-in-first-out.
|
---|
660 | /// This queue automatically resizes itself, so there's no concern of the queue becoming 'full'. Null values are allowed.
|
---|
661 | /// Returns true if the node was successfully enqueued; false if it already exists.
|
---|
662 | /// O(log n)
|
---|
663 | /// </summary>
|
---|
664 | public bool EnqueueWithoutDuplicates(TItem item) {
|
---|
665 | IList<SimpleNode> nodes;
|
---|
666 | if (item == null) {
|
---|
667 | if (_nullNodesCache.Count > 0) {
|
---|
668 | return false;
|
---|
669 | }
|
---|
670 | nodes = _nullNodesCache;
|
---|
671 | } else if (_itemToNodesCache.ContainsKey(item)) {
|
---|
672 | return false;
|
---|
673 | } else {
|
---|
674 | nodes = new List<SimpleNode>();
|
---|
675 | _itemToNodesCache[item] = nodes;
|
---|
676 | }
|
---|
677 | SimpleNode node = EnqueueNoCache(item);
|
---|
678 | nodes.Add(node);
|
---|
679 | return true;
|
---|
680 | }
|
---|
681 |
|
---|
682 | /// <summary>
|
---|
683 | /// Removes an item from the queue. The item does not need to be the head of the queue.
|
---|
684 | /// If the item is not in the queue, an exception is thrown. If unsure, check Contains() first.
|
---|
685 | /// If multiple copies of the item are enqueued, only the first one is removed.
|
---|
686 | /// O(log n)
|
---|
687 | /// </summary>
|
---|
688 | public void Remove(TItem item) {
|
---|
689 | SimpleNode removeMe;
|
---|
690 | IList<SimpleNode> nodes;
|
---|
691 | if (item == null) {
|
---|
692 | if (_nullNodesCache.Count == 0) {
|
---|
693 | throw new InvalidOperationException("Cannot call Remove() on a node which is not enqueued: " + item);
|
---|
694 | }
|
---|
695 | removeMe = _nullNodesCache[0];
|
---|
696 | nodes = _nullNodesCache;
|
---|
697 | } else {
|
---|
698 | if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
699 | throw new InvalidOperationException("Cannot call Remove() on a node which is not enqueued: " + item);
|
---|
700 | }
|
---|
701 | removeMe = nodes[0];
|
---|
702 | if (nodes.Count == 1) {
|
---|
703 | _itemToNodesCache.Remove(item);
|
---|
704 | }
|
---|
705 | }
|
---|
706 | _queue.Remove(removeMe);
|
---|
707 | nodes.Remove(removeMe);
|
---|
708 | }
|
---|
709 |
|
---|
710 | /// <summary>
|
---|
711 | /// Call this method to change the priority of an item.
|
---|
712 | /// Calling this method on a item not in the queue will throw an exception.
|
---|
713 | /// If the item is enqueued multiple times, only the first one will be updated.
|
---|
714 | /// (If your requirements are complex enough that you need to enqueue the same item multiple times <i>and</i> be able
|
---|
715 | /// to update all of them, please wrap your items in a wrapper class so they can be distinguished).
|
---|
716 | /// O(log n)
|
---|
717 | /// </summary>
|
---|
718 | public void UpdatePriority(TItem item) {
|
---|
719 | SimpleNode updateMe = GetExistingNode(item);
|
---|
720 | if (updateMe == null) {
|
---|
721 | throw new InvalidOperationException("Cannot call UpdatePriority() on a node which is not enqueued: " + item);
|
---|
722 | }
|
---|
723 | _queue.UpdatePriority(updateMe);
|
---|
724 | }
|
---|
725 |
|
---|
726 | #region Try* methods for multithreading
|
---|
727 | /// Get the head of the queue, without removing it (use TryDequeue() for that).
|
---|
728 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and First
|
---|
729 | /// Returns true if successful, false otherwise
|
---|
730 | /// O(1)
|
---|
731 | public bool TryFirst(out TItem first) {
|
---|
732 | if (_queue.Count <= 0) {
|
---|
733 | first = default(TItem);
|
---|
734 | return false;
|
---|
735 | }
|
---|
736 |
|
---|
737 | first = _queue.First.Data;
|
---|
738 | return true;
|
---|
739 | }
|
---|
740 |
|
---|
741 | /// <summary>
|
---|
742 | /// Removes the head of the queue (node with minimum priority; ties are broken by order of insertion), and sets it to first.
|
---|
743 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and Dequeue()
|
---|
744 | /// Returns true if successful; false if queue was empty
|
---|
745 | /// O(log n)
|
---|
746 | /// </summary>
|
---|
747 | public bool TryDequeue(out TItem first) {
|
---|
748 | if (_queue.Count <= 0) {
|
---|
749 | first = default(TItem);
|
---|
750 | return false;
|
---|
751 | }
|
---|
752 |
|
---|
753 | SimpleNode node = _queue.Dequeue();
|
---|
754 | first = node.Data;
|
---|
755 | RemoveFromNodeCache(node);
|
---|
756 | return true;
|
---|
757 | }
|
---|
758 |
|
---|
759 | /// <summary>
|
---|
760 | /// Attempts to remove an item from the queue. The item does not need to be the head of the queue.
|
---|
761 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and Remove()
|
---|
762 | /// Returns true if the item was successfully removed, false if it wasn't in the queue.
|
---|
763 | /// If multiple copies of the item are enqueued, only the first one is removed.
|
---|
764 | /// O(log n)
|
---|
765 | /// </summary>
|
---|
766 | public bool TryRemove(TItem item) {
|
---|
767 | SimpleNode removeMe;
|
---|
768 | IList<SimpleNode> nodes;
|
---|
769 | if (item == null) {
|
---|
770 | if (_nullNodesCache.Count == 0) {
|
---|
771 | return false;
|
---|
772 | }
|
---|
773 | removeMe = _nullNodesCache[0];
|
---|
774 | nodes = _nullNodesCache;
|
---|
775 | } else {
|
---|
776 | if (!_itemToNodesCache.TryGetValue(item, out nodes)) {
|
---|
777 | return false;
|
---|
778 | }
|
---|
779 | removeMe = nodes[0];
|
---|
780 | if (nodes.Count == 1) {
|
---|
781 | _itemToNodesCache.Remove(item);
|
---|
782 | }
|
---|
783 | }
|
---|
784 | _queue.Remove(removeMe);
|
---|
785 | nodes.Remove(removeMe);
|
---|
786 | return true;
|
---|
787 | }
|
---|
788 |
|
---|
789 | /// <summary>
|
---|
790 | /// Call this method to change the priority of an item.
|
---|
791 | /// Useful for multi-threading, where the queue may become empty between calls to Contains() and UpdatePriority()
|
---|
792 | /// If the item is enqueued multiple times, only the first one will be updated.
|
---|
793 | /// (If your requirements are complex enough that you need to enqueue the same item multiple times <i>and</i> be able
|
---|
794 | /// to update all of them, please wrap your items in a wrapper class so they can be distinguished).
|
---|
795 | /// Returns true if the item priority was updated, false otherwise.
|
---|
796 | /// O(log n)
|
---|
797 | /// </summary>
|
---|
798 | public bool TryUpdatePriority(TItem item) {
|
---|
799 | SimpleNode updateMe = GetExistingNode(item);
|
---|
800 | if (updateMe == null) {
|
---|
801 | return false;
|
---|
802 | }
|
---|
803 | _queue.UpdatePriority(updateMe);
|
---|
804 | return true;
|
---|
805 | }
|
---|
806 | #endregion
|
---|
807 |
|
---|
808 | public IEnumerator<TItem> GetEnumerator() {
|
---|
809 | return _queue.Select(x => x.Data).GetEnumerator();
|
---|
810 | }
|
---|
811 |
|
---|
812 | IEnumerator IEnumerable.GetEnumerator() {
|
---|
813 | return GetEnumerator();
|
---|
814 | }
|
---|
815 |
|
---|
816 | public bool IsValidQueue() {
|
---|
817 | // Check all items in cache are in the queue
|
---|
818 | foreach (IList<SimpleNode> nodes in _itemToNodesCache.Values) {
|
---|
819 | foreach (SimpleNode node in nodes) {
|
---|
820 | if (!_queue.Contains(node)) {
|
---|
821 | return false;
|
---|
822 | }
|
---|
823 | }
|
---|
824 | }
|
---|
825 |
|
---|
826 | // Check all items in queue are in cache
|
---|
827 | foreach (SimpleNode node in _queue) {
|
---|
828 | if (GetExistingNode(node.Data) == null) {
|
---|
829 | return false;
|
---|
830 | }
|
---|
831 | }
|
---|
832 |
|
---|
833 | // Check queue structure itself
|
---|
834 | return _queue.IsValidQueue();
|
---|
835 | }
|
---|
836 | }
|
---|
837 | } |
---|