PAT Insertion or Heap Sort

PAT Insertion or Heap Sort

根据维基百科：

插入排序迭代，每次重复消耗一个输入元素，并增长一个排序的输出列表。每次迭代，插入排序从输入数据中删除一个元素，在排序列表中找到它所属的位置，并将其插入那里。它重复直到没有输入元素。

堆排序将其输入分为已排序和未排序区域，并通过提取最大元素并将其移动到已排序区域来迭代地缩小未排序区域。它涉及使用堆数据结构而不是线性时间搜索来找到最大值。

现在给定整数的初始序列，以及由某种排序方法多次迭代的结果的序列，你能说出我们使用的是哪种排序方法吗？

输入规格：
每个输入文件包含一个测试用例。对于每种情况，第一行给出一个正整数N (≤100 )。然后在下一行，N个整数作为初始序列给出。最后一行包含部分排序的序列N个数字。假设目标序列总是上升的。一行中的所有数字都用空格分隔。

输出规格：
对于每个测试用例，在第一行打印“插入排序”或“堆排序”以指示用于获得部分结果的方法。然后再运行此方法进行一次迭代，并在第二行中输出结果序列。保证每个测试用例的答案都是唯一的。一行中的所有数字必须用空格分隔，并且行尾不能有多余的空格。

样本输入 1：
10
3 1 2 8 7 5 9 4 6 0
1 2 3 7 8 5 9 4 6 0
样本输出 1：
Insertion Sort
1 2 3 5 7 8 9 4 6 0
样本输入 2：
10
3 1 2 8 7 5 9 4 6 0
6 4 5 1 0 3 2 7 8 9
样本输出 2：
Heap Sort
5 4 3 1 0 2 6 7 8 9

#include "bits/stdc++.h"

using namespace std;

int n;
vector<int> a;
vector<int> b;

void heap_sort(int root, int mx) {
	int left = root * 2 + 1;
	if(left >= mx) return;
	int right = root * 2 + 2;
	
	int j = left;
	if(right < mx && a[left] < a[right]) j = right;
	
	if(a[root] < a[j]) {
		swap(a[root], a[j]);
		heap_sort(j, mx);
	}
}

void insert_sort(int idx) {
	
	for(int i = idx; i >= 1; i--) {
		if(b[i] < b[i - 1]) {
			swap(b[i], b[i - 1]);
		}
		else {
			break;
		}
	}
	
}

int main() {
    
    cin>>n;
    
    vector<int> v(n);
    for(int i = 0; i < n; i++) {
    	int d;
    	cin>>d;
    	a.push_back(d);
	}
	for(int i = 0; i < n; i++) {
		cin>>v[i];
	}
	b = a;
	for(int i = n - 1; i >= 0; i--) {
		heap_sort(i, n);
	}
	bool f = false;
	for(int i = n - 1; i >= 1; i--) {
		swap(a[0], a[i]);
		heap_sort(0, i); 
		
		if(f) {
			cout<<"Heap Sort"<<endl;
			for(int i = 0; i < n; i++) {
				if(i == 0) cout<<a[i];
				else cout<<" "<<a[i];
			}
			cout<<endl;
			return 0;
		}
		if(a == v) f = true;
	}
	f = false;
	for(int i = 1; i < n; i++) {
		insert_sort(i);
		
		if(f) {
			cout<<"Insertion Sort"<<endl;
			for(int i = 0; i < n; i++) {
				if(i == 0) cout<<b[i];
				else cout<<" "<<b[i];
			}
			cout<<endl;
			return 0;
		}
		if(b == v) {
			f = true;
		}
	}
    
    return 0;
}