树查找

[编程题]树查找

热度指数：12717 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 64M，其他语言128M
算法知识视频讲解

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

输入描述:

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

输出描述:

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

示例1

输入

4
1 2 3 4
2

输出

2 3

算法知识视频讲解

Python

华科平凡

python 5行解法

while True:
    try:
        a, b, c, start = int(input()), list(map(int, input().split())), int(input()), 0
        for i in range(c - 1):
            start += 2 ** i
        res = b[start:start + 2 ** (c - 1)]
        print(" ".join(map(str, res)) if res else "EMPTY")

    except:
        break

发表于 2017-10-01 15:09:12 回复(2)