GCC Code Coverage Report
Directory: . Exec Total Coverage
File: src/prop/minisat/mtl/Heap.h Lines: 69 77 89.6 %
Date: 2021-11-08 Branches: 35 124 28.2 %

Line Exec Source
1
/******************************************************************************************[Heap.h]
2
Copyright (c) 2003-2006, Niklas Een, Niklas Sorensson
3
Copyright (c) 2007-2010, Niklas Sorensson
4
5
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
6
associated documentation files (the "Software"), to deal in the Software without restriction,
7
including without limitation the rights to use, copy, modify, merge, publish, distribute,
8
sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
9
furnished to do so, subject to the following conditions:
10
11
The above copyright notice and this permission notice shall be included in all copies or
12
substantial portions of the Software.
13
14
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT
15
NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
16
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
17
DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT
18
OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
19
**************************************************************************************************/
20
21
#ifndef Minisat_Heap_h
22
#define Minisat_Heap_h
23
24
#include "base/check.h"
25
#include "prop/minisat/mtl/Vec.h"
26
27
namespace cvc5 {
28
namespace Minisat {
29
30
//=================================================================================================
31
// A heap implementation with support for decrease/increase key.
32
33
34
template<class Comp>
35
30674
class Heap {
36
    Comp     lt;       // The heap is a minimum-heap with respect to this comparator
37
    vec<int> heap;     // Heap of integers
38
    vec<int> indices;  // Each integers position (index) in the Heap
39
40
    // Index "traversal" functions
41
81160926
    static inline int left  (int i) { return i*2+1; }
42
72596964
    static inline int right (int i) { return (i+1)*2; }
43
99947174
    static inline int parent(int i) { return (i-1) >> 1; }
44
45
46
51199014
    void percolateUp(int i)
47
    {
48
51199014
        int x  = heap[i];
49
51199014
        int p  = parent(i);
50
51
148695334
        while (i != 0 && lt(x, heap[p])){
52
48748160
            heap[i]          = heap[p];
53
48748160
            indices[heap[p]] = i;
54
48748160
            i                = p;
55
48748160
            p                = parent(p);
56
        }
57
51199014
        heap   [i] = x;
58
51199014
        indices[x] = i;
59
51199014
    }
60
61
62
9592325
    void percolateDown(int i)
63
    {
64
9592325
        int x = heap[i];
65
55734003
        while (left(i) < heap.size()){
66
30408207
            int child = right(i) < heap.size() && lt(heap[right(i)], heap[left(i)]) ? right(i) : left(i);
67
30408207
            if (!lt(heap[child], x)) break;
68
23070839
            heap[i]          = heap[child];
69
23070839
            indices[heap[i]] = i;
70
23070839
            i                = child;
71
        }
72
9592325
        heap   [i] = x;
73
9592325
        indices[x] = i;
74
9592325
    }
75
76
77
  public:
78
30684
    Heap(const Comp& c) : lt(c) { }
79
80
1175
    int  size      ()          const { return heap.size(); }
81
7487663
    bool empty     ()          const { return heap.size() == 0; }
82
210261961
    bool inHeap    (int n)     const { return n < indices.size() && indices[n] >= 0; }
83
    int operator[](int index) const
84
    {
85
      Assert(index < heap.size());
86
      return heap[index];
87
    }
88
89
42046101
    void decrease(int n)
90
    {
91
42046101
      Assert(inHeap(n));
92
42046101
      percolateUp(indices[n]);
93
42046101
    }
94
694813
    void increase(int n)
95
    {
96
694813
      Assert(inHeap(n));
97
694813
      percolateDown(indices[n]);
98
694813
    }
99
100
    // Safe variant of insert/decrease/increase:
101
429303
    void update(int n)
102
    {
103
429303
        if (!inHeap(n))
104
1948
            insert(n);
105
        else {
106
427355
            percolateUp(indices[n]);
107
427355
            percolateDown(indices[n]); }
108
429303
    }
109
110
111
8725558
    void insert(int n)
112
    {
113
8725558
        indices.growTo(n+1, -1);
114
8725558
        Assert(!inHeap(n));
115
116
8725558
        indices[n] = heap.size();
117
8725558
        heap.push(n);
118
8725558
        percolateUp(indices[n]);
119
8725558
    }
120
121
122
7388548
    int  removeMin()
123
    {
124
7388548
        int x            = heap[0];
125
7388548
        heap[0]          = heap.last();
126
7388548
        indices[heap[0]] = 0;
127
7388548
        indices[x]       = -1;
128
7388548
        heap.pop();
129
7388548
        if (heap.size() > 1) percolateDown(0);
130
7388548
        return x;
131
    }
132
133
134
    // Rebuild the heap from scratch, using the elements in 'ns':
135
22499
    void build(vec<int>& ns) {
136
2551814
        for (int i = 0; i < heap.size(); i++)
137
2529315
            indices[heap[i]] = -1;
138
22499
        heap.clear();
139
140
2318564
        for (int i = 0; i < ns.size(); i++){
141
2296065
            indices[ns[i]] = i;
142
2296065
            heap.push(ns[i]); }
143
144
1166503
        for (int i = heap.size() / 2 - 1; i >= 0; i--)
145
1144004
            percolateDown(i);
146
22499
    }
147
148
    void clear(bool dealloc = false)
149
    {
150
        for (int i = 0; i < heap.size(); i++)
151
            indices[heap[i]] = -1;
152
        heap.clear(dealloc);
153
    }
154
};
155
156
157
//=================================================================================================
158
}
159
}  // namespace cvc5
160
161
#endif