GCC Code Coverage Report
Directory: . Exec Total Coverage
File: src/prop/minisat/mtl/Heap.h Lines: 69 77 89.6 %
Date: 2021-09-29 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
12650
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
43656433
    static inline int left  (int i) { return i*2+1; }
42
39356864
    static inline int right (int i) { return (i+1)*2; }
43
27480225
    static inline int parent(int i) { return (i-1) >> 1; }
44
45
46
11588683
    void percolateUp(int i)
47
    {
48
11588683
        int x  = heap[i];
49
11588683
        int p  = parent(i);
50
51
43371767
        while (i != 0 && lt(x, heap[p])){
52
15891542
            heap[i]          = heap[p];
53
15891542
            indices[heap[p]] = i;
54
15891542
            i                = p;
55
15891542
            p                = parent(p);
56
        }
57
11588683
        heap   [i] = x;
58
11588683
        indices[x] = i;
59
11588683
    }
60
61
62
5779441
    void percolateDown(int i)
63
    {
64
5779441
        int x = heap[i];
65
31146541
        while (left(i) < heap.size()){
66
16232184
            int child = right(i) < heap.size() && lt(heap[right(i)], heap[left(i)]) ? right(i) : left(i);
67
16232184
            if (!lt(heap[child], x)) break;
68
12683550
            heap[i]          = heap[child];
69
12683550
            indices[heap[i]] = i;
70
12683550
            i                = child;
71
        }
72
5779441
        heap   [i] = x;
73
5779441
        indices[x] = i;
74
5779441
    }
75
76
77
  public:
78
12656
    Heap(const Comp& c) : lt(c) { }
79
80
919
    int  size      ()          const { return heap.size(); }
81
3162043
    bool empty     ()          const { return heap.size() == 0; }
82
62494450
    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
7330172
    void decrease(int n)
90
    {
91
7330172
      Assert(inHeap(n));
92
7330172
      percolateUp(indices[n]);
93
7330172
    }
94
1572217
    void increase(int n)
95
    {
96
1572217
      Assert(inHeap(n));
97
1572217
      percolateDown(indices[n]);
98
1572217
    }
99
100
    // Safe variant of insert/decrease/increase:
101
690822
    void update(int n)
102
    {
103
690822
        if (!inHeap(n))
104
13073
            insert(n);
105
        else {
106
677749
            percolateUp(indices[n]);
107
677749
            percolateDown(indices[n]); }
108
690822
    }
109
110
111
3580762
    void insert(int n)
112
    {
113
3580762
        indices.growTo(n+1, -1);
114
3580762
        Assert(!inHeap(n));
115
116
3580762
        indices[n] = heap.size();
117
3580762
        heap.push(n);
118
3580762
        percolateUp(indices[n]);
119
3580762
    }
120
121
122
3109954
    int  removeMin()
123
    {
124
3109954
        int x            = heap[0];
125
3109954
        heap[0]          = heap.last();
126
3109954
        indices[heap[0]] = 0;
127
3109954
        indices[x]       = -1;
128
3109954
        heap.pop();
129
3109954
        if (heap.size() > 1) percolateDown(0);
130
3109954
        return x;
131
    }
132
133
134
    // Rebuild the heap from scratch, using the elements in 'ns':
135
12060
    void build(vec<int>& ns) {
136
1049037
        for (int i = 0; i < heap.size(); i++)
137
1036977
            indices[heap[i]] = -1;
138
12060
        heap.clear();
139
140
906640
        for (int i = 0; i < ns.size(); i++){
141
894580
            indices[ns[i]] = i;
142
894580
            heap.push(ns[i]); }
143
144
457399
        for (int i = heap.size() / 2 - 1; i >= 0; i--)
145
445339
            percolateDown(i);
146
12060
    }
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