Создал(а) 'a3/program.c'
This commit is contained in:
parent
298918db19
commit
8da21f1839
188
a3/program.c
Normal file
188
a3/program.c
Normal file
@ -0,0 +1,188 @@
|
||||
// Max binárna kopa
|
||||
#include <assert.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
// Ľubovoľné (dynamické) pole môže byť binárna kopa
|
||||
struct heap {
|
||||
int* array;
|
||||
int size;
|
||||
int capacity;
|
||||
};
|
||||
|
||||
// Výpočet rodičovského uzla
|
||||
int parent(int i){
|
||||
return (i -1) / 2;
|
||||
}
|
||||
|
||||
// Výpočeť ľavého syna
|
||||
int left_child(int i){
|
||||
return (2*i) + 1;
|
||||
}
|
||||
|
||||
// Výpočet pravého syna
|
||||
int right_child(int i){
|
||||
return (2*i) + 2;
|
||||
}
|
||||
|
||||
// Výmena dvoch prvkov v poli
|
||||
void swap(int* array,int a, int b){
|
||||
int c = array[a];
|
||||
array[a] = array[b];
|
||||
array[b] = c;
|
||||
}
|
||||
|
||||
// Test kopovitosti
|
||||
// Pozeráme, či je každý potomok menší ako jeho rodič
|
||||
// Funkcia zlyhá, ak nájdeme aspoň jedného potomka, väčšieho alebo rovného ako je rodič
|
||||
void check_heap_property(struct heap* h){
|
||||
for (int i = 0; i < h->size / 2 ; i++){
|
||||
int c = left_child(i);
|
||||
if (c >= h->size){
|
||||
break;
|
||||
}
|
||||
assert(h->array[c] >= h->array[i]);
|
||||
c = right_child(i);
|
||||
if (c >= h->size){
|
||||
break;
|
||||
}
|
||||
assert(h->array[c] >= h->array[i]);
|
||||
}
|
||||
}
|
||||
//
|
||||
// Vytvorenie prázdnej kopy
|
||||
struct heap* create_heap(int capacity){
|
||||
struct heap* h = malloc(sizeof(struct heap));
|
||||
h->array = calloc(capacity,sizeof(int));
|
||||
h->capacity = capacity;
|
||||
h->size = 0;
|
||||
return h;
|
||||
}
|
||||
|
||||
// Zmazanie ľubovoľnej kopy
|
||||
void delete_heap(struct heap* h){
|
||||
assert(h);
|
||||
free(h->array);
|
||||
free(h);
|
||||
}
|
||||
|
||||
// Recursive heapification
|
||||
/**
|
||||
* Predpokladáme, že ľavý a pravý podstrom prvku i sú kopy.
|
||||
*
|
||||
* Výsledkom je, že strom s vrcholom i je kopa.
|
||||
*
|
||||
* @param pole
|
||||
* @param velkost pola
|
||||
* @param Koren stromu, ktorý sa má upraviť
|
||||
*/
|
||||
void heapify2(int* array,int size,int i){
|
||||
// Zoberiem i-ty prvok
|
||||
int largest = i;
|
||||
int l = left_child(i);
|
||||
int r = right_child(i);
|
||||
// Pozrieme jeho hodnotu a hodnoty jeho potomkov
|
||||
// Zistíme kde v tejto trojici sa nachádza najväčšia hodnota
|
||||
if (l < size && array[largest] < array[l]){
|
||||
largest = l;
|
||||
}
|
||||
if (r < size && array[largest] < array[r]){
|
||||
largest = r;
|
||||
}
|
||||
// Ak je najväčšia hodnota v potomkovi
|
||||
if (largest != i){
|
||||
// Vymeníme ich hodnoty
|
||||
int v = array[i];
|
||||
array[i] = array[largest];
|
||||
array[largest] = v;
|
||||
// Obnovíme kopovitosť v potomkovi
|
||||
heapify2(array,size, largest);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
///////////////////////////////////////////
|
||||
void heapify_array(int* array, int size){
|
||||
// Začíname od predposlednej úrovne
|
||||
// sme si istí, že všetky prvky na poslednej
|
||||
// úrovni sú binárne kopy (lebo nemajú potomkov)
|
||||
// Postupne tak obnovíme kopovitosť celého poľa
|
||||
for (int i = size / 2; i >= 0; i-- ){
|
||||
heapify2(array,size,i);
|
||||
}
|
||||
}
|
||||
////////////////////////////////////////////
|
||||
|
||||
|
||||
|
||||
int remove_max(struct heap* h){
|
||||
assert(h->size > 0);
|
||||
int i = 0;
|
||||
int res = h->array[i];
|
||||
// Vymeníme prvý a posledný prvok a skrátime pole
|
||||
h->array[i] = h->array[h->size -1];
|
||||
h->size -= 1;
|
||||
// Obnonovíme kopovitosť
|
||||
heapify2(h->array,h->size,0);
|
||||
return res;
|
||||
}
|
||||
|
||||
|
||||
|
||||
// Pridanie do kopy
|
||||
void add(struct heap* h,int value){
|
||||
// Pridáme prvok na koniec poľa
|
||||
int i = h->size - 1;
|
||||
h->size += 1;
|
||||
assert(h->size <= h->capacity);
|
||||
h->array[i] = value;
|
||||
// Pozrieme na jeho rodiča
|
||||
int p = parent(i);
|
||||
// Pokiaľ rodič existuje a jeho hodnota je menšia
|
||||
while(p >= 0 && h->array[p] < h->array[i]){
|
||||
// Vymeníme ich hodnoty
|
||||
int v = h->array[p];
|
||||
h->array[p] = h->array[i];
|
||||
h->array[i] = v;
|
||||
// Pokračujeme ďalším rodičom
|
||||
i = p;
|
||||
p = parent(p);
|
||||
}
|
||||
}
|
||||
|
||||
/***
|
||||
* Z kopy vytvorí zotriedené pole
|
||||
*/
|
||||
void heap_sort(int* heap,int size){
|
||||
// Na začiatku je najväčší prvok
|
||||
for (int i = size -1; i >= 0; i -= 2){
|
||||
// Vymenime prvy a posledny prvok
|
||||
int v = heap[0];
|
||||
heap[0] = heap[i];
|
||||
heap[i] = v;
|
||||
// Na konci je najväčší prvok.
|
||||
// Uvažujeme, že kopa je o jedno menšia
|
||||
// Obnovíme kopovitosť
|
||||
heapify2(heap,i,0);
|
||||
}
|
||||
}
|
||||
|
||||
int main(){
|
||||
struct heap* kopa = create_heap(10);
|
||||
add(kopa,10);
|
||||
add(kopa,20);
|
||||
add(kopa,40);
|
||||
add(kopa,15);
|
||||
for (int i = 0; i < kopa->size; i++){
|
||||
printf("%d ",kopa->array[i]);
|
||||
}
|
||||
printf("\n");
|
||||
int pole[14] = {2,3,1,4,5,3,5,6,7,4,7,4,3,3};
|
||||
int sz = 14;
|
||||
heapify_array(pole,sz);
|
||||
for (int i = 0; i < sz; i++){
|
||||
printf("%d ",pole[i]);
|
||||
}
|
||||
printf("\n");
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user