一個最大堆的實現(最基本的功能)

參考了算法導論上的堆排序算法,實現瞭如下最大堆(採用的數據結構爲數組)。只有最基本的功能,最後利用該最大堆實現了堆排序(非遞減排序)。代碼如下:

max_heap_sort.h:

#define PARENT(i) ((i - 1)/2)
#define LEFT(i) (2*i + 1)
#define RIGHT(i) (2*i + 2)

void max_heapify(vector<int> & vec_ints, int heap_size, int i);
void build_max_heap(vector<int> & vec_ints, int heap_size);
void max_heap_sort(vector<int> & vec_ints);

max_heap_sort.cpp:

void max_heapify(vector<int> & vec_ints, int heap_size, int i)
{
	int largest = -1;
	int l = LEFT(i);
	int r = RIGHT(i);

	if (l < heap_size && vec_ints[l] > vec_ints[i])
	{
		largest = l;
	} 
	else
	{
		largest = i;
	}

	if (r < heap_size && vec_ints[r] > vec_ints[largest])
	{
		largest = r;
	}

	if (largest != i)
	{
		swap(vec_ints[i], vec_ints[largest]);
		max_heapify(vec_ints, heap_size, largest);
	}
}

void build_max_heap(vector<int> & vec_ints, int heap_size)
{
	int s = PARENT(heap_size - 1);
	for (int i = s; i >=0; i--)
	{
		max_heapify(vec_ints, heap_size, i);
	}
}

void max_heap_sort(vector<int> & vec_ints)
{
	int heap_size = vec_ints.size();
	build_max_heap(vec_ints, heap_size);
	for (int i = vec_ints.size() - 1; i > 0; i--)
	{
		swap(vec_ints[i], vec_ints[0]);
		heap_size--;
		max_heapify(vec_ints, heap_size, 0);
	}
}


發佈了26 篇原創文章 · 獲贊 4 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章