summaryrefslogtreecommitdiff
path: root/libs/minisat/Heap.h
diff options
context:
space:
mode:
authorClifford Wolf <clifford@clifford.at>2014-03-12 10:17:51 +0100
committerClifford Wolf <clifford@clifford.at>2014-03-12 10:17:51 +0100
commit94c1307c262e4b14f4a91b1bbcf9099ee6202bab (patch)
treeaae192666509ba533b9a9640679a55b002cd869d /libs/minisat/Heap.h
parent9087ece97c20d76359fb23cfd7a0f13552c1f2fd (diff)
Added libs/minisat (copy of minisat git master)
Diffstat (limited to 'libs/minisat/Heap.h')
-rw-r--r--libs/minisat/Heap.h168
1 files changed, 168 insertions, 0 deletions
diff --git a/libs/minisat/Heap.h b/libs/minisat/Heap.h
new file mode 100644
index 00000000..a7512462
--- /dev/null
+++ b/libs/minisat/Heap.h
@@ -0,0 +1,168 @@
+/******************************************************************************************[Heap.h]
+Copyright (c) 2003-2006, Niklas Een, Niklas Sorensson
+Copyright (c) 2007-2010, Niklas Sorensson
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
+associated documentation files (the "Software"), to deal in the Software without restriction,
+including without limitation the rights to use, copy, modify, merge, publish, distribute,
+sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or
+substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT
+NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
+DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT
+OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+**************************************************************************************************/
+
+#ifndef Minisat_Heap_h
+#define Minisat_Heap_h
+
+#include "libs/minisat/Vec.h"
+#include "libs/minisat/IntMap.h"
+
+namespace Minisat {
+
+//=================================================================================================
+// A heap implementation with support for decrease/increase key.
+
+
+template<class K, class Comp, class MkIndex = MkIndexDefault<K> >
+class Heap {
+ vec<K> heap; // Heap of Keys
+ IntMap<K,int,MkIndex> indices; // Each Key's position (index) in the Heap
+ Comp lt; // The heap is a minimum-heap with respect to this comparator
+
+ // Index "traversal" functions
+ static inline int left (int i) { return i*2+1; }
+ static inline int right (int i) { return (i+1)*2; }
+ static inline int parent(int i) { return (i-1) >> 1; }
+
+
+ void percolateUp(int i)
+ {
+ K x = heap[i];
+ int p = parent(i);
+
+ while (i != 0 && lt(x, heap[p])){
+ heap[i] = heap[p];
+ indices[heap[p]] = i;
+ i = p;
+ p = parent(p);
+ }
+ heap [i] = x;
+ indices[x] = i;
+ }
+
+
+ void percolateDown(int i)
+ {
+ K x = heap[i];
+ while (left(i) < heap.size()){
+ int child = right(i) < heap.size() && lt(heap[right(i)], heap[left(i)]) ? right(i) : left(i);
+ if (!lt(heap[child], x)) break;
+ heap[i] = heap[child];
+ indices[heap[i]] = i;
+ i = child;
+ }
+ heap [i] = x;
+ indices[x] = i;
+ }
+
+
+ public:
+ Heap(const Comp& c, MkIndex _index = MkIndex()) : indices(_index), lt(c) {}
+
+ int size () const { return heap.size(); }
+ bool empty () const { return heap.size() == 0; }
+ bool inHeap (K k) const { return indices.has(k) && indices[k] >= 0; }
+ int operator[](int index) const { assert(index < heap.size()); return heap[index]; }
+
+ void decrease (K k) { assert(inHeap(k)); percolateUp (indices[k]); }
+ void increase (K k) { assert(inHeap(k)); percolateDown(indices[k]); }
+
+
+ // Safe variant of insert/decrease/increase:
+ void update(K k)
+ {
+ if (!inHeap(k))
+ insert(k);
+ else {
+ percolateUp(indices[k]);
+ percolateDown(indices[k]); }
+ }
+
+
+ void insert(K k)
+ {
+ indices.reserve(k, -1);
+ assert(!inHeap(k));
+
+ indices[k] = heap.size();
+ heap.push(k);
+ percolateUp(indices[k]);
+ }
+
+
+ void remove(K k)
+ {
+ assert(inHeap(k));
+
+ int k_pos = indices[k];
+ indices[k] = -1;
+
+ if (k_pos < heap.size()-1){
+ heap[k_pos] = heap.last();
+ indices[heap[k_pos]] = k_pos;
+ heap.pop();
+ percolateDown(k_pos);
+ }else
+ heap.pop();
+ }
+
+
+ K removeMin()
+ {
+ K x = heap[0];
+ heap[0] = heap.last();
+ indices[heap[0]] = 0;
+ indices[x] = -1;
+ heap.pop();
+ if (heap.size() > 1) percolateDown(0);
+ return x;
+ }
+
+
+ // Rebuild the heap from scratch, using the elements in 'ns':
+ void build(const vec<K>& ns) {
+ for (int i = 0; i < heap.size(); i++)
+ indices[heap[i]] = -1;
+ heap.clear();
+
+ for (int i = 0; i < ns.size(); i++){
+ // TODO: this should probably call reserve instead of relying on it being reserved already.
+ assert(indices.has(ns[i]));
+ indices[ns[i]] = i;
+ heap.push(ns[i]); }
+
+ for (int i = heap.size() / 2 - 1; i >= 0; i--)
+ percolateDown(i);
+ }
+
+ void clear(bool dispose = false)
+ {
+ // TODO: shouldn't the 'indices' map also be dispose-cleared?
+ for (int i = 0; i < heap.size(); i++)
+ indices[heap[i]] = -1;
+ heap.clear(dispose);
+ }
+};
+
+
+//=================================================================================================
+}
+
+#endif