堆排序_heap_sort

#include <iostream>
#include <vector>
#include <conio.h>

using namespace std;

class DataType:public vector<int>
{
public: 
	int heapSize;
	DataType():heapSize(0){};
	~DataType(){};
protected:
private:
};

/*
	将index为结点的子树调整为最大堆，保持最大堆的性质，以index的左右孩子为根的二叉树必须也是最大堆。
*/
void max_heapify(DataType &data, int index)
{
	int l = index << 1;		//计算左孩子
	int r = l + 1;			//得到右孩子
	
	//找出该节点和其左右孩子中最大的
	int largest = 0;
	if (l <= data.heapSize && data[l] > data[index])
	{
		largest = l;
	}
	else
	{
		largest = index;
	}

	if (r <= data.heapSize && data[r] > data[largest])
	{
		largest = r;
	}

	//若该节点不为最大，则与最大的交换，并在最大的节点处递归调用该函数。
	if (largest != index)
	{
		int temp = data[index];
		data[index] = data[largest];
		data[largest] = temp;

		max_heapify(data, largest);
	}
}

/*
	建立最大堆
*/
void buile_max_heap(DataType &data)
{
	//因为为了方便，data[0]为设定的0，所以要减一
	data.heapSize = data.size() - 1;	
	for (int i = data.heapSize / 2; i > 0; i--)
	{
		max_heapify(data, i);
	}

}

/*
	堆排序算法
*/
void heap_sort(DataType &data)
{
	//建立最大堆
	buile_max_heap(data);
	//将最大值交换到最后，并调用max_heapify使堆保持最大堆的性质
	for (int i = data.size() - 1; i > 1; i--)
	{
		int temp = data[i];
		data[i] = data[1];
		data[1] = temp;

		data.heapSize--;
		max_heapify(data, 1);
	}
}

int main()
{
	DataType data;
	data.push_back(0);
	int temp = 0;
	while (cin >> temp)
	{
		data.push_back(temp);
	}

	heap_sort(data);

	for (vector<int>::iterator iter = data.begin(); iter != data.end(); iter++)
	{
		cout << *iter;
		cout << " ";
	}
	getch();
	return 0;
}