heap build
This commit is contained in:
parent
0c862db205
commit
48464840ba
72
heap.cpp
72
heap.cpp
|
@ -1,6 +1,5 @@
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <cmath>
|
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
|
|
||||||
|
|
||||||
|
@ -10,40 +9,56 @@ public:
|
||||||
void push(int);
|
void push(int);
|
||||||
void print() const;
|
void print() const;
|
||||||
bool empty() const;
|
bool empty() const;
|
||||||
|
|
||||||
|
Heap(std::vector<int> v);
|
||||||
private:
|
private:
|
||||||
std::vector<int> data;
|
std::vector<int> data;
|
||||||
|
|
||||||
|
void bubble_up(int index);
|
||||||
|
void bubble_down(int index);
|
||||||
void dump(std::ofstream&, int) const;
|
void dump(std::ofstream&, int) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
void Heap::push(int h){
|
Heap::Heap(std::vector<int> v): data(std::move(v)){
|
||||||
data.push_back(h);
|
|
||||||
for(int i = data.size(); i > 1; i /= 2){
|
for(int i = data.size()/2; i >= 0; i--){
|
||||||
|
bubble_down(i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void Heap::bubble_up(int index){
|
||||||
|
for(int i = index; i > 1; i /= 2){//neindexuje od 1, jen nulty prvek nema kam vysplhat
|
||||||
if(data.at(i-1) < data.at(i/2-1)) std::swap(data[i-1], data[(i/2)-1]);
|
if(data.at(i-1) < data.at(i/2-1)) std::swap(data[i-1], data[(i/2)-1]);
|
||||||
else break;
|
else break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Heap::bubble_down(int index){
|
||||||
|
while(true){
|
||||||
|
if(data.size() <= 2*index+1) break;
|
||||||
|
int small_child_i;
|
||||||
|
if(data.size() > 2*index+2 && data.at(2*index+2) < data.at(2*index+1))
|
||||||
|
small_child_i = 2*index+2;
|
||||||
|
else small_child_i = 2*index+1;
|
||||||
|
|
||||||
|
if(data.at(index) > data.at(small_child_i)){
|
||||||
|
std::swap(data[index], data[small_child_i]);
|
||||||
|
index = small_child_i;
|
||||||
|
}else break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void Heap::push(int h){
|
||||||
|
data.push_back(h);
|
||||||
|
bubble_up(data.size());
|
||||||
|
}
|
||||||
|
|
||||||
int Heap::pop(){
|
int Heap::pop(){
|
||||||
if (empty()) throw std::out_of_range("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
|
if (empty()) throw std::out_of_range("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
|
||||||
std::swap(data[0], data[data.size()-1]);
|
std::swap(data[0], data[data.size()-1]);
|
||||||
int garbage = data.back();
|
int garbage = data.back();
|
||||||
data.pop_back();
|
data.pop_back();
|
||||||
|
bubble_down(0);
|
||||||
int muj_index = 0;
|
|
||||||
while(true){
|
|
||||||
if(data.size() <= 2*muj_index+1) break;
|
|
||||||
int small_child_i;
|
|
||||||
if(data.size() > 2*muj_index+2 && data.at(2*muj_index+2) < data.at(2*muj_index+1))
|
|
||||||
small_child_i = 2*muj_index+2;
|
|
||||||
else small_child_i = 2*muj_index+1;
|
|
||||||
|
|
||||||
if(data.at(muj_index) > data.at(small_child_i)){
|
|
||||||
std::swap(data[muj_index], data[small_child_i]);
|
|
||||||
muj_index = small_child_i;
|
|
||||||
}else break;
|
|
||||||
}
|
|
||||||
|
|
||||||
return garbage;
|
return garbage;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -77,17 +92,16 @@ void Heap::print() const {
|
||||||
|
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
Heap h;
|
Heap h({9, 7, 6, 5, 4, 3, 2, 1, 0, -3, 8});
|
||||||
std::vector<int> v = {1, 4, 9, 0, 7, 5, 8};
|
|
||||||
|
|
||||||
for (auto i : v) {
|
// for (auto i : v) {
|
||||||
h.push(i);
|
// h.push(i);
|
||||||
h.print();
|
h.print();
|
||||||
}
|
// }
|
||||||
|
|
||||||
while (!h.empty()){
|
// while (!h.empty()){
|
||||||
int i = h.pop();
|
// int i = h.pop();
|
||||||
h.print();
|
// h.print();
|
||||||
}
|
// }
|
||||||
int aaaaaaaaaaaaaaa = h.pop();
|
// int aaaaaaaaaaaaaaa = h.pop();
|
||||||
}
|
}
|
Loading…
Reference in New Issue