【题解】【AcWing】1640. 堆

158 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

1640. 堆

原题传送:AcWing 1640. 堆

在计算机科学中,堆是一种的基于树的专用数据结构,它具有堆属性:

如果 PPCC 的父结点,则在大顶堆中 PP 结点的权值大于或等于 CC 结点的权值,在小顶堆中 PP 结点的权值小于或等于 CC 结点的权值。

一种堆的常见实现是二叉堆,它是由完全二叉树来实现的。

你的任务是判断给定的完全二叉树是否是堆。

输入格式

第一行包含两个整数 MMNN ,分别表示给定完全二叉树的数量以及每个完全二叉树包含的结点数量。

接下来 MM 行,每行包含 NN 个不同的整数(都在int范围内),表示一个完全二叉树的层序遍历序列。

输出格式

对于每个给定的二叉树,首先输出一行对它是否是堆的判断结论。

如果是大顶堆,则输出MaxHeap,如果是小顶堆,则输出MinHeap,如果不是堆,则输出NotHeap

然后,再输出一行它的后序遍历序列。

同行数字用空格隔开,行首行尾不得有多余空格。

数据范围

1M1001 \le M \le 100 , 1<N10001<N \le 1000

输入样例:

3 8
98 72 86 60 65 12 23 50
8 38 25 58 52 82 70 60
10 28 15 12 34 9 8 56

输出样例:

Max Heap
50 60 65 72 12 23 86 98
Min Heap
60 58 52 38 82 70 25 8
Not Heap
56 12 34 28 9 8 15 10

思路:

根据父结点大于或者小于子结点判断是否为大小根堆,既有大于又有小于的不是堆。

题解:

#include<bits/stdc++.h>

using namespace std;

const int N = 1010;

int n;
int h[N];

void dfs(int u)
{
	if(u * 2 <= n)
		dfs(u * 2);
	if(u * 2 + 1 <= n)
		dfs(u * 2 + 1);
		
	printf("%d", h[u]);
	if(u != 1)
		printf(" ");
}

int main()
{	
	int T;
	scanf("%d%d", &T, &n);

	while(T--)
	{
		for(int i = 1; i <= n; i++)
			scanf("%d", &h[i]);
		
		bool lt = false, gt = false;
		for(int i = 1; i <= n; i++)
		{
			for(int j = 0; j < 2; j++)
			{
				if(i * 2 + j <= n)
				{
					int a = h[i], b = h[i * 2 + j];
					if(a < b)
						lt = true;
					else
						gt = true;
				}
			}
		}
		if(lt && gt)
			printf("Not Heap\n");
		else if(lt)
			printf("Min Heap\n");
		else
			printf("Max Heap\n");
			
		dfs(1);
		printf("\n");
	}
    
	return 0;
}