코딩뚠뚠

[기본문제풀이] heap_sort 본문

알고리즘 문제풀이/기본문제풀이

[기본문제풀이] heap_sort

로디네로 2020. 12. 27. 15:20
반응형

 

풀이 일시 : 2020-08-04

힙정렬 :

힙 트리구조를 이용하는 정렬방법이다.

힙정렬을 하려면 정해진 데이터를 힙구조를 가지도록 만들어야 한다.

( 힙은 최솟값이나 최댓값을 빠르게 찾아내기 위해 완전 이진트리를 기반으로 하는 트리이다.

최대힙은 부모노드가 자식노드보다 큰 힙이다. )

 

힙생성 알고리즘 - 큰 자식과 자신의 위치를 바꾸는 것이다. 즉 O(logN)

전체 트리를 힙구조로 만드는 복잡도는 결국 O(N*logN)

이를 가지고 루트의 값을 가장 뒤쪽으로 보내면서 힙트리의 크기를 1씩 빼준다.

가장 뒤쪽으로 보낸 것은 정렬이 된 것이고 다시 나머지에 대해 힙생성 알고리즘을 수행해준다.

결국 전체 힙 정렬의 복잡도는 O(N*logN)

문제를 보면서 이해해보도록 하자.

문제 :

7 6 5 8 3 5 9 1 6 을 힙정렬로 오름차순정렬하라

 

풀이 : 

#include <stdio.h>

int number = 9;
int heap[9] = { 7,6,5,8,3,5,9,1,6 };

int main(void) {

	//힙 생성하는 알고리즘
	for (int i = 1; i < number; i++) {
		int c = i;
		do {
			int root = (c - 1) / 2;
			if (heap[root] < heap[c]) { //최대힙 생성위해 SWAP해준다.
				int temp = heap[root];
				heap[root] = heap[c];
				heap[c] = temp;
			}
			c = root;
		} while (c != 0);
	}

	// 하나씩 뒤로보내고 다시 힙을 생성하는 알고리즘
	for (int i = number - 1; i >= 0; i--) {
		// 루트의 값을 맨뒤로 보내는데 그 맨뒤 라는것은 하나씩 줄어든다.
		int temp = heap[0];
		heap[0] = heap[i];
		heap[i] = temp;
		int root = 0; //루트는 항상 0일것
		int c = 1; //힙을 다시 구성하기 위한 변수
		//크기에 맞게 힙을 생성해줌
		do {
			c = 2 * root + 1; //c는 root의 자식이 될것(자식의 왼쪽값)
			//자식중에 더 큰 자식을 찾는것
			if (c < i - 1 && heap[c] < heap[c + 1]) {//오른쪽값이 크다면 오른쪽값을 가리킴
				c++;
			}
			//루트보다 자식이 더 크다면 교환한다.
			if (c < i && heap[root] < heap[c]) {
				temp = heap[root];
				heap[root] = heap[c];
				heap[c] = temp;
			}
			root = c; //이제 그 자식이 root가 될것. 그리고 반복
		} while (c < i); //child가 하나씩 작아지는 i(최대값)보다 크게될경우 중단
	}
	for (int i = 0; i < number; i++) {
		printf("%d", heap[i]);
	}
}
반응형