从键盘输入二叉树的元素,建立二叉树,实现二叉树的遍历算法。 【基本要求】 实现以下基本操作: (1) 从键盘输入二叉树的元素,建立二叉树。 (2) 实现二叉树的先序遍历算法。非递归算法
时间: 2023-12-01 09:14:24 浏览: 23
好的,我可以帮您解答这个问题。
首先,我们需要定义二叉树的节点类,包括节点值,左右子节点。代码如下:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
```
接下来,我们需要实现从键盘输入二叉树的元素,建立二叉树的操作。我们可以通过递归的方式实现,对于每个节点,先输入它的值,然后递归输入它的左右子节点。代码如下:
```python
def buildTree():
val = input("请输入节点值:")
if not val:
return None
root = TreeNode(int(val))
root.left = buildTree()
root.right = buildTree()
return root
```
接下来,我们需要实现二叉树的先序遍历算法。非递归算法可以通过栈来实现,具体实现步骤如下:
1. 初始化栈,将根节点入栈。
2. 循环执行以下操作,直到栈为空:
- 弹出栈顶节点并访问。
- 如果该节点存在右子节点,将右子节点入栈。
- 如果该节点存在左子节点,将左子节点入栈。
代码如下:
```python
def preorderTraversal(root: TreeNode):
if not root:
return []
stack = [root]
res = []
while stack:
cur = stack.pop()
res.append(cur.val)
if cur.right:
stack.append(cur.right)
if cur.left:
stack.append(cur.left)
return res
```
这样,我们就实现了从键盘输入二叉树的元素,建立二叉树,以及二叉树的先序遍历算法的非递归实现。完整代码如下:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def buildTree():
val = input("请输入节点值:")
if not val:
return None
root = TreeNode(int(val))
root.left = buildTree()
root.right = buildTree()
return root
def preorderTraversal(root: TreeNode):
if not root:
return []
stack = [root]
res = []
while stack:
cur = stack.pop()
res.append(cur.val)
if cur.right:
stack.append(cur.right)
if cur.left:
stack.append(cur.left)
return res
if __name__ == "__main__":
root = buildTree()
print(preorderTraversal(root))
```
希望可以帮到您!