PAT 甲级 1155 Heap Paths 30分

题目

原题链接：
题目描述

解析

判断一串数，是否构成大顶堆、小顶堆或不是堆
并按从右子树到左子树的深度优先遍历顺序输出所有遍历的情况

代码

#include
#include
using namespace std;

vector<int> heap;
int n;
int a[1001];
int isMin = 1, isMax = 1;

void dfs(int index){
	if(index*2>n && index*2+1>n){
	//判断是否到达叶子节点
		if(index<=n){
			for(int i=0;i<heap.size();i++){
				printf("%d%s",heap[i],i==heap.size()-1?"\n":" ");
			}
		}
	}else{
		//右子树入栈进行深度遍历
		heap.push_back(a[index*2+1]);
		dfs(index*2+1);
		heap.pop_back();
		//左子树入栈进行深度遍历
		heap.push_back(a[index*2]);
		dfs(index*2);	
		heap.pop_back();	
	}
}

int main(){
//	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	heap.push_back(a[1]);
	//有一个违反大/小顶堆的情况，则将对应的flag修改为0
	for(int i=2;i<=n;i++){
		if(a[i/2]>a[i]) isMin=0;
		if(a[i/2]<a[i]) isMax=0;	
	}
	dfs(1);
	if(isMin == 1) printf("Min Heap\n");
	else if(isMax == 1) printf("Max Heap\n");
	else printf("Not Heap\n");
	return 0;
}