1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
|
# Практикум №9 по Структури от данни, спец. ИС
## `Binary Heap`
**Binary Heap**-a представлява дървовидна структура от данни, която изпълнява 2 условия:
1) пълно двоично дърво е (всички нива са запълнени, освен последното);
2) всеки връх има стойност <= (респективно >=) на тази на родителя си (heap property);
> [!IMPORTANT]
> От условията следва, че винаги най-големият (респективно най-малкият) елемент е на върха на пирамидата, НО данните вътре не са винаги сортирани.
Обикновено се представя чрез масив, като е изпълнено следното:
- за всеки родител на индекс i лявото му дете е на индекс 2i + 1, а дясното - на 2i + 2;
- за всяко дете на индекс i родителят му се намира на индекс (i - 1) / 2;
- коренът е на индекс 0.
Разграничаваме 2 вида heap:
- `minHeap` - най-малкият елемент е най-отгоре;
- `maxHeap` - най-големият елемент е най-отгоре;
### Как от вектор можем да създадем heap?
STL поддържа методи, които ни позволяват да работим с вектор, като го превърнем в heap. Това са следните:
- **Превръщане на heap**: `std::make_heap(begin_it, end_it)` - O(N)
- **Добавяне на елемент**:
- Добавяме елемент в края на контейнера.
- Викаме `std::push_heap(begin, end)` - O(logN)
- **Извличане на най-големия елемент**:
- Викаме `std::pop_heap(begin, end)` - O(logN)
- Премахваме елемента в края.
- **Проверка дали е heap**: `std::is_heap(begin, end)` - O(N)
Нека разгледаме и техните имплементации:
```c++
#include <iostream>
#include <vector>
template <typename RandomIt, typename Compare>
void heapify(RandomIt first, RandomIt last, RandomIt root, const Compare& comp) {
size_t size = std::distance(first, last);
size_t rootIdx = std::distance(first, root);
while (true) {
size_t largest = rootIdx;
size_t leftChild = 2 * rootIdx + 1;
size_t rightChild = 2 * rootIdx + 2;
if (leftChild < size && comp(*(first + largest), *(first + leftChild)))
largest = leftChild;
if (rightChild < size&& comp(*(first + largest), *(first + rightChild)))
largest = rightChild;
if (largest == rootIdx)
break;
std::swap(*(first + rootIdx), *(first + largest));
rootIdx = largest;
}
}
template <typename RandomIt, typename Compare = std::less<>>
void make_heap(RandomIt first, RandomIt last, Compare comp = Compare()) {
size_t size = std::distance(first, last);
for (int i = (size / 2); i >= 0; --i)
heapify(first, last, first + i, comp);
}
template <typename RandomIt, typename Compare = std::less<>>
void pop_heap(RandomIt first, RandomIt last, Compare comp = Compare()) {
size_t size = std::distance(first, last);
if (size > 1) {
std::swap(*first, *(last - 1));
heapify(first, last, first, comp);
}
}
template <typename RandomIt, typename Compare>
void bubble_up(RandomIt first, RandomIt last, const Compare& comp) {
int childIndex = std::distance(first, last) - 1;
while (childIndex > 0) {
int parentIndex = (childIndex - 1) / 2;
auto parent = first + parentIndex;
if (!comp(*parent, *(first + childIndex)))
break;
std::swap(*parent, *(first + childIndex));
childIndex = parentIndex;
}
}
template <typename RandomIt, typename Compare = std::less<>>
void push_heap(RandomIt first, RandomIt last, Compare comp = Compare()) {
size_t size = std::distance(first, last);
if (size > 1)
bubble_up(first, last, comp);
}
int main()
{
std::vector<int> v = { 1, 9, 2, 5, 3, 15, 6, 98, 10 };
make_heap(v.begin(), v.end());
for (auto i : v)
std::cout << i << " ";
pop_heap(v.begin(), v.end());
for (auto i : v)
std::cout << i << " ";
push_heap(v.begin(), v.end());
for (auto i : v)
std::cout << i << " ";
}
```
Както се вижда, по default методите използват `std::less<>` за сравнение на обектите, така че default-ния heap, който ще създадем, е `maxHeap`.
> [!NOTE]
> Ако искаме да създадем `minHeap`, то като comparator трябва да подадем `std::greater<>`.
Структурата от данни, която е създадена на базата на `Binary Heap`-a, е т.нар. `Priority Queue` или приоритетна опашка на български.
## `std::priority_queue`
`std::priority_queue` представлява обвиващ клас за контейнер (std::vector по default), изграден на основата на heap-a. По default е изграден като maxHeap, т.е.
ни гарантира, че винаги най-големия елемент ще е първи. Поддържа следните операции:
- **Добавяне на елемент**: `pq.push(value)` - O(logN)
- **Премахване на елемента с най-висок приоритет**: `pq.pop()` - O(logN)
- **Достъп до елемента с най-висок приоритет**: `pq.top()` - O(1)
- **Проверка на размера**: `pq.size()` - O(1)
- **Проверка за празнота**: `pq.empty()` - O(1)
Нека разгледаме какви параметри приема в шаблона си:
```c++
template<
class T,
class Container = std::vector<T>,
class Compare = std::less<typename Container::value_type>
> class priority_queue;
```
Ето и примерна имплементация:
```c++
#include <iostream>
#include <vector>
#include <algorithm>
template<class T, class Container = std::vector<T>, class Comp = std::less<T>>
class priority_queue
{
Container _c;
Comp _comp;
public:
void push(const T& value)
{
_c.push_back(value);
std::push_heap(_c.begin(), _c.end(), _comp);
}
void push(T&& value)
{
_c.push_back(std::move(value));
std::push_heap(_c.begin(), _c.end(), _comp);
}
void pop()
{
std::pop_heap(_c.begin(), _c.end(), _comp);
_c.pop_back();
}
bool empty() const
{
return _c.empty();
}
const T& top() const
{
if (empty())
throw std::runtime_error("Empty container");
return _c.front();
}
size_t size() const
{
return _c.size();
}
};
```
И тук, ако искаме да създадем minHeap, трябва да подадем `std::greater<>` като custom comparator.
### Пример за създаване на min-heap
```c++
#include <queue>
#include <vector>
#include <iostream>
int main()
{
std::vector<int> v = {3, 1, 4, 1, 5, 9};
std::priority_queue<int, std::vector<int>, std::greater<int>> pq(v.begin(), v.end());
while (!pq.empty())
{
std::cout << pq.top() << ' ';
pq.pop();
}
}
```
---
[**Линк към задачите**](https://leetcode.com/problem-list/aglkjy9v/)
|