-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathheap.c
51 lines (37 loc) · 949 Bytes
/
heap.c
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
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
void swap(int *a, int *b) {
int tmp = *a;
*a = *b;
*b = tmp;
}
#define LEFT(i) ((i) * 2 + 1)
#define RIGHT(i) ((i) * 2 + 2)
void heapify(int arr[], int item, int size) {
int min = item;
if (LEFT(item) < size && arr[LEFT(item)] < arr[min]) min = LEFT(item);
if (RIGHT(item) < size && arr[RIGHT(item)] < arr[min]) min = RIGHT(item);
if (item != min) {
swap(arr + item, arr + min);
heapify(arr, min, size);
}
}
void heap_sort(int arr[], int size) {
for (int i = size / 2; i >= 0; i--)
heapify(arr, i, size);
for (int i = size - 1; i > 0; i--) {
swap(arr, arr + i);
heapify(arr, 0, i - 1);
}
}
int main(int argc, char* argv[]) {
if (argc < 2) return 1;
int size = atoi(argv[1]);
int *arr = (int*)malloc(sizeof(int) * size);
for (int i = 0; i < size; i++)
scanf("%d", arr + i);
heap_sort(arr, size);
free(arr);
return 0;
}