从键盘上输入一组数据,建立一个以二叉链表形式存储的二叉树。对所建立的二叉树分别进行先序、中序和后序遍历,并输出结果。
时间: 2024-03-20 15:40:00 浏览: 62
好的,我来解答你的问题。
以下是建立二叉树、先序遍历、中序遍历和后序遍历的代码,你可以参考一下:
```python
# 定义二叉树节点
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
# 建立二叉树
def build_tree():
print("请输入节点值(空格分隔,输入 # 表示空节点):")
vals = input().split()
if not vals:
return None
root = TreeNode(int(vals[0]))
queue = [root]
index = 1
while index < len(vals):
node = queue.pop(0)
if vals[index] != "#":
node.left = TreeNode(int(vals[index]))
queue.append(node.left)
index += 1
if index < len(vals) and vals[index] != "#":
node.right = TreeNode(int(vals[index]))
queue.append(node.right)
index += 1
return root
# 先序遍历
def preorder_traversal(root):
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
res.append(node.val)
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
return res
# 中序遍历
def inorder_traversal(root):
if not root:
return []
res = []
stack = []
node = root
while stack or node:
while node:
stack.append(node)
node = node.left
node = stack.pop()
res.append(node.val)
node = node.right
return res
# 后序遍历
def postorder_traversal(root):
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
res.append(node.val)
if node.left:
stack.append(node.left)
if node.right:
stack.append(node.right)
return res[::-1]
# 主函数
if __name__ == "__main__":
root = build_tree()
print("先序遍历:", preorder_traversal(root))
print("中序遍历:", inorder_traversal(root))
print("后序遍历:", postorder_traversal(root))
```
你可以将以上代码保存到一个.py文件中,然后运行该文件,在命令行中输入一组数据即可建立二叉树并遍历。
阅读全文