PAT (Advanced Level) Practice 1147 Heaps

本题仅仅用到堆的性质（大顶堆和小顶堆）并不涉及对堆的代码考察，实际上考察的还是完全二叉树的性质，以及通过层序遍历得到后序遍历
之前也认真写过一次，但是没有写出来；这次又试了试，过了
对于遍历，还是先记录过程中的数据；最后在main()函数里输出结果

#include <cstdio>
#include <vector>
using namespace std;
const int maxn=1e3+10;
int n, layer[maxn];
vector<int> v;
bool judgemax(){
	for(int i=1; i<=n; i++){
		if(2*i<=n){
			if(layer[2*i]>layer[i])
			    return false;
		} 
		
		if(2*i+1<=n){
			if(layer[2*i+1]>layer[i])
			    return false;
		}
	}
	
	return true;
}

bool judgemin(){
	for(int i=1; i<=n; i++){
	    if(2*i<=n){
			if(layer[2*i]<layer[i])
			    return false;
		} 
		
		if(2*i+1<=n){
			if(layer[2*i+1]<layer[i])
			    return false;
		}
	}
	
	return true;
}

void dfs(int root){
	if(root>n)
	    return;
	
	dfs(2*root);    
	dfs(2*root+1);
	v.push_back(layer[root]);
}

int main()
{
	int m;
	scanf("%d%d", &m, &n);
	while(m--){
		v.clear();
		for(int i=1; i<=n; i++)
	    scanf("%d", &layer[i]);
	    
	    if(judgemax()){
		    printf("Max Heap\n");
		    dfs(1);
		    for(int i=0; i<v.size(); i++){
			    printf("%d", v[i]);
			    printf(i+1!=v.size() ? " ": "\n");
		    }
		
	    } else if(judgemin()){
		    printf("Min Heap\n");
		    dfs(1);
		    for(int i=0; i<v.size(); i++){
			    printf("%d", v[i]);
			    printf(i+1!=v.size() ? " ": "\n");
		    }
		
	    } else {
		    printf("Not Heap\n");
		    dfs(1);
	 	    for(int i=0; i<v.size(); i++){
			    printf("%d", v[i]);
			    printf(i+1!=v.size() ? " ": "\n");
		     }
	    }
	}
	
	return 0;
}