1.二叉树的构建以及层次遍历(即广度优先遍历)(完全二叉树)
class Node(object):
def __init__(self, item):
self.elem = item
self.lchild = None
self.rchild = None
class Tree(object):
def __init__(self):
self.root = None
def add(self, item):
node = Node(item)
if self.root is None:
self.root = node
return
queue = [self.root]
while queue:
cur_node = queue.pop(0)
if cur_node.lchild is None:
cur_node.lchild = node
return
else:
queue.append(cur_node.lchild)
if cur_node.rchild is None:
cur_node.rchild = node
return
else:
queue.append(cur_node.rchild)
def bianli(self):
if self.root is None:
return
queue = [self.root]
while queue:
cur_node = queue.pop(0)
print(cur_node.elem)
if cur_node.lchild :
queue.append(cur_node.lchild)
if cur_node.rchild :
queue.append(cur_node.rchild)
tree = Tree()
tree.add(1)
tree.add(2)
tree.add(3)
tree.add(4)
tree.bianli()
2.深度遍历(三种形式)
def preorder(node):
if node is None:
return
print(node.elem)
preorder(node.lchild)
preorder(node.rchild)
tree = Tree()
tree.add(1)
tree.add(2)
tree.add(3)
tree.add(4)
tree.add(5)
tree.add(6)
tree.add(7)
preorder(tree.root)
def inorder(node):
if node is None:
return
preorder(node.lchild)
print(node.elem)
preorder(node.rchild)
tree = Tree()
tree.add(1)
tree.add(2)
tree.add(3)
tree.add(4)
tree.add(5)
tree.add(6)
tree.add(7)
inorder(tree.root)
def posorder(node):
if node is None:
return
preorder(node.lchild)
preorder(node.rchild)
print(node.elem)
tree = Tree()
tree.add(1)
tree.add(2)
tree.add(3)
tree.add(4)
tree.add(5)
tree.add(6)
tree.add(7)
posorder(tree.root)