GCC Code Coverage Report
Directory: . Exec Total Coverage
File: src/prop/minisat/mtl/Heap.h Lines: 69 77 89.6 %
Date: 2021-05-21 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
17976
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
63309500
    static inline int left  (int i) { return i*2+1; }
42
57496842
    static inline int right (int i) { return (i+1)*2; }
43
52204368
    static inline int parent(int i) { return (i-1) >> 1; }
44
45
46
24612539
    void percolateUp(int i)
47
    {
48
24612539
        int x  = heap[i];
49
24612539
        int p  = parent(i);
50
51
79796197
        while (i != 0 && lt(x, heap[p])){
52
27591829
            heap[i]          = heap[p];
53
27591829
            indices[heap[p]] = i;
54
27591829
            i                = p;
55
27591829
            p                = parent(p);
56
        }
57
24612539
        heap   [i] = x;
58
24612539
        indices[x] = i;
59
24612539
    }
60
61
62
7288793
    void percolateDown(int i)
63
    {
64
7288793
        int x = heap[i];
65
45527983
        while (left(i) < heap.size()){
66
23729972
            int child = right(i) < heap.size() && lt(heap[right(i)], heap[left(i)]) ? right(i) : left(i);
67
23729972
            if (!lt(heap[child], x)) break;
68
19119595
            heap[i]          = heap[child];
69
19119595
            indices[heap[i]] = i;
70
19119595
            i                = child;
71
        }
72
7288793
        heap   [i] = x;
73
7288793
        indices[x] = i;
74
7288793
    }
75
76
77
  public:
78
17976
    Heap(const Comp& c) : lt(c) { }
79
80
1053
    int  size      ()          const { return heap.size(); }
81
4495907
    bool empty     ()          const { return heap.size() == 0; }
82
116292465
    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
18674219
    void decrease(int n)
90
    {
91
18674219
      Assert(inHeap(n));
92
18674219
      percolateUp(indices[n]);
93
18674219
    }
94
1577883
    void increase(int n)
95
    {
96
1577883
      Assert(inHeap(n));
97
1577883
      percolateDown(indices[n]);
98
1577883
    }
99
100
    // Safe variant of insert/decrease/increase:
101
691357
    void update(int n)
102
    {
103
691357
        if (!inHeap(n))
104
13073
            insert(n);
105
        else {
106
678284
            percolateUp(indices[n]);
107
678284
            percolateDown(indices[n]); }
108
691357
    }
109
110
111
5260036
    void insert(int n)
112
    {
113
5260036
        indices.growTo(n+1, -1);
114
5260036
        Assert(!inHeap(n));
115
116
5260036
        indices[n] = heap.size();
117
5260036
        heap.push(n);
118
5260036
        percolateUp(indices[n]);
119
5260036
    }
120
121
122
4429467
    int  removeMin()
123
    {
124
4429467
        int x            = heap[0];
125
4429467
        heap[0]          = heap.last();
126
4429467
        indices[heap[0]] = 0;
127
4429467
        indices[x]       = -1;
128
4429467
        heap.pop();
129
4429467
        if (heap.size() > 1) percolateDown(0);
130
4429467
        return x;
131
    }
132
133
134
    // Rebuild the heap from scratch, using the elements in 'ns':
135
16233
    void build(vec<int>& ns) {
136
1488479
        for (int i = 0; i < heap.size(); i++)
137
1472246
            indices[heap[i]] = -1;
138
16233
        heap.clear();
139
140
1292736
        for (int i = 0; i < ns.size(); i++){
141
1276503
            indices[ns[i]] = i;
142
1276503
            heap.push(ns[i]); }
143
144
651657
        for (int i = heap.size() / 2 - 1; i >= 0; i--)
145
635424
            percolateDown(i);
146
16233
    }
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