设计一个程序,输入一颗二叉树,根据结点数据找出该结点对应的所有孩子结点的完整代码
时间: 2024-05-15 20:17:56 浏览: 26
请注意:以下代码仅供参考,可能需要根据具体情况进行修改。
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def find_children(root:TreeNode, val:int)->list:
result = []
queue = [root]
while queue:
node = queue.pop(0)
if node.val == val:
if node.left:
result.append(node.left.val)
if node.right:
result.append(node.right.val)
return result
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return result
```
使用示例:
```python
# 构建一个二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)
# 查找结点2的孩子结点
print(find_children(root, 2))
# 输出:[4, 5]
# 查找结点3的孩子结点
print(find_children(root, 3))
# 输出:[6, 7]
# 查找结点4的孩子结点
print(find_children(root, 4))
# 输出:[](没有孩子结点)
# 查找结点8的孩子结点
print(find_children(root, 8))
# 输出:[](未找到结点)
```