Создал(а) 'a3/program.c'

This commit is contained in:
Oleksandr Hryshchenko 2021-11-22 16:46:40 +00:00
parent 298918db19
commit 8da21f1839

188
a3/program.c Normal file
View 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 kopy.
*
* Výsledkom je, že strom s vrcholom i je kopa.
*
* @param pole
* @param velkost pola
* @param Koren stromu, ktorý sa 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;
}