不忘初心,砥砺前行!
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;
}