zcmu-4932: 树查找【完全二叉树+输出指定层数节点】

不忘初心，砥砺前行！

4932: 树查找

Time Limit: 1 Sec  Memory Limit: 32 MB
Submit: 6  Solved: 4
[Submit][Status][Web Board]

Description

有一棵树，输出某一深度的所有节点，有则输出这些节点，无则输出EMPTY。该树是完全二叉树。

Input

输入有多组数据。
每组输入一个n(1<=n<=1000)，然后将树中的这n个节点依次输入，再输入一个d代表深度。

Output

输出该树中第d层得所有节点，节点间用空格隔开，最后一个节点后没有空格。

Sample Input

5 1 2 3 4 5 7 7 1 2 3 4 5 6 7 2 0

Sample Output

EMPTY 2 3

HINT

Source

数据结构高分笔记

解题思路：其实这题不用什么bfs，这题说了是完全二叉树，我们只要找到完全二叉树与数组存储的关系就好了。用数组存储是从0开始，然后它的左孩子就是2*i+1，右孩子是2*i+2，然后自己在草稿纸上画出这棵二叉树，把位置的关系对应上去，然后找出它们之间的联系就好了。我把它分成3种情况，具体的见代码吧。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
int a[1001];
using namespace std;
int main(void)
{
	int n;
	while(~scanf("%d",&n)&&n)
	{
		memset(a,0,sizeof a);
		for(int i=0;i<n;i++) scanf("%d",&a[i]);
		int b=(int)(log(n)/log(2));
		int d;
		scanf("%d",&d);
		if(d>b+1)
		{
			printf("EMPTY\n");
			continue;
		}
		else if(d==b+1)
		{
		    int i=(int)pow(2,d-1)-1;
		    printf("%d",a[i]);
		    for(i=i+1;i<n;i++) printf(" %d",a[i]);
		    cout<<endl;
		}
		else
		{
			int i=(int)pow(2,d-1)-1;
			int j=(int)pow(2,d)-1;
		    printf("%d",a[i]);
		    for(i=i+1;i<j;i++) printf(" %d",a[i]);
		    cout<<endl;
		}
	}
	return 0;
}