SCAU 8644 堆排序

8644 堆排序
时间限制:1000MS 代码长度限制:10KB
提交次数:1909 通过次数:1257

题型: 编程题 语言: G++;GCC
Description
用函数实现堆排序，并输出每趟排序的结果

输入格式
第一行：键盘输入待排序关键的个数n
第二行：输入n个待排序关键字，用空格分隔数据

输出格式
第一行：初始建堆后的结果
其后各行输出交换堆顶元素并调整堆的结果，数据之间用一个空格分隔

输入样例
10
5 4 8 0 9 3 2 6 7 1

输出样例
9 7 8 6 4 3 2 5 0 1
8 7 3 6 4 1 2 5 0 9
7 6 3 5 4 1 2 0 8 9
6 5 3 0 4 1 2 7 8 9
5 4 3 0 2 1 6 7 8 9
4 2 3 0 1 5 6 7 8 9
3 2 1 0 4 5 6 7 8 9
2 0 1 3 4 5 6 7 8 9
1 0 2 3 4 5 6 7 8 9
0 1 2 3 4 5 6 7 8 9

代码：

#include <iostream>
#include<vector>
#include<algorithm>
using namespace std;
void heapify(int arr[], int n, int i)//调整堆,i为父节点
{
	int largest = i;
	//largest为父节点
	//先调整，依据完全二叉树性质
	int l = 2 * largest + 1;//数组下标从0开始，故要加1
	int r = l + 1;

	//更新父节点的索引
	if (l < n&&arr[largest] < arr[l])largest = l;
	if (r<n&&arr[largest] < arr[r])largest = r;

	// 如果最大元素不是当前父节点，则进行交换操作，并递归调整堆结构
	if (largest != i) {
		swap(arr[i], arr[largest]);
		heapify(arr, n, largest); 
	}

}
void heapsort(int arr[],int n)//建堆
{
	//构建初始大根堆
	//从最后一个非叶子节点开始，依次向上调整堆。这里最后一个非叶子节点的位置是 n/2-1，其中n为数组长度。
	for (int i = n / 2 - 1;i >= 0;i--)heapify(arr,n,i);

	//依次取出堆顶元素交换至末尾，然后将剩余元素重新调整为大根堆
	for (int i = n - 1;i >= 0;i--)//注意数组下标是从0开始
	{
		for (int i = 0;i < n;i++)cout << arr[i]<<" ";//输出的是每个大根堆构建成的序列，故在交换前输出
		cout<<endl;
		swap(arr[i], arr[0]);
		heapify(arr, i, 0);
	}
}
int main()
{
	int n;
	cin >> n;
	int* arr = new int[n + 5];
	for (int i = 0;i < n;i++)cin >> arr[i];
	heapsort(arr, n);
	
	return 0;
}