HeapSort
힙 소트(Heap Sort)
---
완전 이진 트리를 기본으로 하는 힙(Heap) 자료구조를 기반으로한 정렬 방식
완전 이진 트리란?
> 삽입할 때 왼쪽부터 차례대로 추가하는 이진 트리
힙 소트는 `불안정 정렬`에 속함
시간복잡도
| 평균 | 최선 | 최악 | | :------: | :------: | :------: | | Θ(nlogn) | Ω(nlogn) | O(nlogn) |
- 과정
1. 최대 힙을 구성 2. 현재 힙 루트는 가장 큰 값이 존재함. 루트의 값을 마지막 요소와 바꾼 후, 힙의 사이즈 하나 줄임 3. 힙의 사이즈가 1보다 크면 위 과정 반복
<img src="https://t1.daumcdn.net/cfile/tistory/999896445AD4953023">
루트를 마지막 노드로 대체 (11 → 4), 다시 최대 힙 구성
<img src="https://t1.daumcdn.net/cfile/tistory/99E1AD445AD4953015">
이와 같은 방식으로 최대 값을 하나씩 뽑아내면서 정렬하는 것이 힙 소트
java
public void heapSort(int[] array) {
int n = array.length;
// max heap 초기화
for (int i = n/2-1; i>=0; i--){
heapify(array, n, i); // 1
}
// extract 연산
for (int i = n-1; i>0; i--) {
swap(array, 0, i);
heapify(array, i, 0); // 2
}
}
- 1번째 heapify
> 일반 배열을 힙으로 구성하는 역할 > > 자식노드로부터 부모노드 비교 > > > > - n/2-1부터 0까지 인덱스가 도는 이유는? > > 부모 노드의 인덱스를 기준으로 왼쪽 자식노드 (i2 + 1), 오른쪽 자식 노드(i2 + 2)이기 때문
- 2번째 heapify
> 요소가 하나 제거된 이후에 다시 최대 힙을 구성하기 위함 > > 루트를 기준으로 진행(extract 연산 처리를 위해)
java
public void heapify(int array[], int n, int i) {
int p = i;
int l = i*2 + 1;
int r = i*2 + 2;
//왼쪽 자식노드
if (l < n && array[p] < array[l]) {
p = l;
}
//오른쪽 자식노드
if (r < n && array[p] < array[r]) {
p = r;
}
//부모노드 < 자식노드
if(i != p) {
swap(array, p, i);
heapify(array, n, p);
}
}
다시 최대 힙을 구성할 때까지 부모 노드와 자식 노드를 swap하며 재귀 진행
퀵정렬과 합병정렬의 성능이 좋기 때문에 힙 정렬의 사용빈도가 높지는 않음.
하지만 힙 자료구조가 많이 활용되고 있으며, 이때 함께 따라오는 개념이 `힙 소트`
- 힙 소트가 유용할 때
- 가장 크거나 가장 작은 값을 구할 때
> 최소 힙 or 최대 힙의 루트 값이기 때문에 한번의 힙 구성을 통해 구하는 것이 가능
- 최대 k 만큼 떨어진 요소들을 정렬할 때
> 삽입정렬보다 더욱 개선된 결과를 얻어낼 수 있음
- 전체 소스 코드
java
private void solve() {
int[] array = { 230, 10, 60, 550, 40, 220, 20 };
heapSort(array);
for (int v : array) {
System.out.println(v);
}
}
public static void heapify(int array[], int n, int i) {
int p = i;
int l = i * 2 + 1;
int r = i * 2 + 2;
if (l < n && array[p] < array[l]) {
p = l;
}
if (r < n && array[p] < array[r]) {
p = r;
}
if (i != p) {
swap(array, p, i);
heapify(array, n, p);
}
}
public static void heapSort(int[] array) {
int n = array.length;
// init, max heap
for (int i = n / 2 - 1; i >= 0; i--) {
heapify(array, n, i);
}
// for extract max element from heap
for (int i = n - 1; i > 0; i--) {
swap(array, 0, i);
heapify(array, i, 0);
}
}
public static void swap(int[] array, int a, int b) {
int temp = array[a];
array[a] = array[b];
array[b] = temp;
}