XDOJ 335 堆排序

思路：建立最小堆 首尾交换 重建最小堆~

堆排序

题目描述：编写堆排序算法，建立小顶堆，测试数据为整数。

输入：

第一行是待排序数据元素的个数（小于100）； 第二行是待排序的数据元素。

输出：

建立小顶堆的结果。

样例输入：

10

50 36 41 19 23 4 20 18 12 22

样例输出：

4 12 20 18 22 41 50 36 19 23

#include<iostream>
#include<vector>
using namespace std;

//构建小顶堆 递归~~~
//len 是 num长度 index是第一个非叶子节点的下标
//确保此节点符合标准
void adjust2min(vector<int> &num, int len, int index) {
	int l = 2 * index + 1;  //左子节点
	int r = 2 * index + 2;  //右子节点

	int minIdx = index;
	if (l < len && num[l] < num[minIdx]) {
		minIdx = l;
	}
	if (r < len && num[r] < num[minIdx]) {
		minIdx = r;
	}
	if (minIdx != index) {
		swap(num[minIdx], num[index]);
		adjust2min(num, len, minIdx);
	}
}

void heapSort(vector<int> &num, int size) {
	//构建小根堆（从最后一个非叶子节点向上）
	for (int i = size / 2 - 1; i >= 0; i--) {
		adjust2min(num, size, i);
	}

	//调整小根堆（题干未要求）
//	for (int i = size - 1; i >= 1; i--) {
//		swap(num[0], num[i]); //尾部最小
//		adjust2min(num, i, 0); //继续堆排序
//	}
}

int main() {
	
	vector<int> num;
	int n;
	cin >> n;
	for (int i = 0; i < n; i++) {
		int k;
		cin >> k;
		num.push_back(k);
	}
	heapSort(num, num.size());
	for (int i = 0; i < n; i++) {
		cout << num[i] << ' ';
	}


}