1155 Heap Paths1155 Heap Paths1155 Heap Paths

题目大意:

判断一个完全二叉树是不是堆,是最大堆还是最小堆,并且从右到左输出它的每一条路径。

解题思路:

DFS遍历+数组记录路径即可。
代码如下:

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
int n,data[1100];
vector<int> v;
bool flagma,flagmi,flagnot;//堆类型 
void dfs(int p)
{
	if(p*2>n)//边界,叶节点 
	{
    	v.push_back(data[p]);//输出路径 
	    printf("%d",v[0]);
	    for(int i=1;i<v.size();i++)printf(" %d",v[i]);
	    cout<<endl;
		return;
	}
	v.push_back(data[p]);
	if((p*2+1)<=n)//先走右边 
	{
		if(data[p]<=data[p*2+1])//判断堆类型 
		{
			if(!flagma)flagmi=true;
			else flagnot=true;//出现冲突 
		}
		else
		{
			if(!flagmi)flagma=true;
			else flagnot=true;
		}
		dfs(p*2+1);
		v.pop_back();
	}
	if(p*2<=n)
	{
		if(data[p]<=data[p*2])
		{
			if(!flagma)flagmi=true;
			else flagnot=true;
		}
		else
		{
			if(!flagmi)flagma=true;
			else flagnot=true;
		}
		dfs(p*2);
		v.pop_back();
	}
	return ;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&data[i]);
	dfs(1);
	if(!flagnot)
	{
		if(flagma)printf("Max Heap\n");
		else printf("Min Heap\n");
	}
	else printf("Not Heap\n");
}

相关文章: