完全二叉树每层元素的查找

4932: 树查找
Time Limit: 1 Sec Memory Limit: 32 MB
[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

数据结构高分笔记
AC_code：
/*
自我感觉写的比较粗劣，但是正在努力提升自己的水平~
*/

    #include <stdio.h>
    #include <string.h>
    #include <math.h>
    int a[1005];
    int main()
    {
        int n;
        while(~scanf("%d",&n)&&n!=0)
        {
            memset(a,0,sizeof(a));
            for(int i = 1; i <= n; i++)
            {
                scanf("%d",&a[i]);
            }
            int d,num = 1,k;
            scanf("%d",&d);
            for( int i = 1; ; i++)
            {
                num *= 2;
                if( num >= n+1)
                {
                    k = i;
                    break;
                }
            }
            if(d > k)
                printf("EMPTY\n");
            else
            {
                int begin_t = (int)pow(2,d-1);
                int t = 1;
                printf("%d",a[begin_t]);
                if(d < k || num == n+1)
                {
                    for(int j = begin_t+1; ; j++)
                    {
                        if(t == begin_t ) break;
                        printf(" %d",a[j]);
                        t++;
                    }
                }
                if( n+1 < num  && d == k)
                {
                    for(int i = begin_t+1; a[i] != 0; i++)
                        printf(" %d",a[i]);
                }
                printf("\n");
            }
        }
        return 0;
    }