题目阐释:
s型打印,重要的是将binary—tree 逐层遍历,获取每层的node。
思路:
将树的遍历转化为 压栈出栈。 每次将列表内的node全部出栈,获取子元素,然后全部再入栈。 如此反复迭代
应用:
当树有层次信息时候,可以如此操作。
代码如下:
class Node(object): def __init__(self, val): self.left_node = None self.right_node = None self.value = valclass MakeNode(object): def __init__(self): pass def make_series(self, root_val, value_list_in): root = Node(root_val) # cur_node=root cur_nodes = list() cur_nodes.append(root) while cur_nodes: nodes_iters = list() while cur_nodes: cur_node = cur_nodes.pop(0) nodes_iters.append(cur_node) for node_iter in nodes_iters: if value_list_in: values_iter = value_list_in.pop(0) if values_iter[0]: node = Node(values_iter[0]) node_iter.left_node = node cur_nodes.append(node) if values_iter[1]: node = Node(values_iter[1]) node_iter.right_node = node cur_nodes.append(node) # print(root) return rootclass BinaryTree(object): def __init__(self): pass def stack(self,root): nodes=[root] flag=0 while nodes: nodes_tmp=list() while nodes: nodes_tmp.append(nodes.pop(0)) # print('nodes_tmp==>',nodes_tmp) if flag%2==0: nodes_print=nodes_tmp else: nodes_print=nodes_tmp[::-1] for val_node in nodes_print: if val_node: print(val_node.value) for node_iter in nodes_tmp: if node_iter: # print(node_iter.value) nodes.extend([node_iter.left_node,node_iter.right_node]) flag+=1if __name__ == '__main__': root = 1 values = [[2,3], [4, 5], [6, 7], [8, 9], [10, 11], [12, None], [None, 13]] mn = MakeNode() root=mn.make_series(root, values) bt=BinaryTree() bt.stack(root)