1098 Insertion or Heap Sort

题目大意

给出一串序列和按“插入排序”或“堆排序”的中间结果，让你判断是哪种方法排的，并按这种方法再排一次。

---

思路解析

本题考察对堆排序的掌握，可见PAT也并不是一成不变的，还是要全面啊。

---

示例代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> vec, v2;
void swap(int i, int j) {
	int temp = vec[i];
	vec[i] = vec[j];
	vec[j] = temp;
}
void heapify(int n, int i) {
	int max = i;
	int c1 = i * 2 + 1;
	int c2 = i * 2 + 2;
	if (c1 < n && vec[c1] > vec[max]) {
		max = c1;
	}
	if (c2 < n && vec[c2] > vec[max]) {
		max = c2;
	}
	if (max != i) {
		swap(i, max);
		heapify(n, max);
	}
}
void build_heap(vector<int>& vec, int n) {
	for (int i = (n - 2) / 2; i >= 0; i--) {
		heapify(n, i);
	}
}
int main() {
	int n;
	scanf("%d", &n);
	vec.resize(n);
	v2.resize(n);
	for (int i = 0; i < n; i++) {
		scanf("%d", &vec[i]);
	}
	for (int i = 0; i < n; i++) {
		scanf("%d", &v2[i]);
	}
	int index = 1;
	for (int i = 1; i < n; i++) {
		if (v2[i] < v2[i - 1]) {
			index = i;
			break;
		}
	}
	if (index != 1) {
		printf("Insertion Sort\n");
		sort(vec.begin(), vec.begin() + index + 1);
	}
	else {//堆排序
		printf("Heap Sort\n");
		bool flag = false;
		build_heap(vec, n);
		for (int i = n - 1; i >= 0; i--) {
			swap(i, 0);
			heapify(i, 0);
			if (flag) break;
			int j = 0;
			for (j; j < n && vec[j] == v2[j]; j++);
			if (j == n) flag = true;
		}
	}
	printf("%d", vec[0]);
	for (int i = 1; i < n; i++) {
		printf(" %d", vec[i]);
	}
	return 0;
}