程序员求职经验分享与学习资料整理平台

网站首页 > 文章精选 正文

python如何表示树,遍历树

balukai 2025-02-13 11:02:56 文章精选 16 ℃

在 Python 中,可以使用类和对象的方式来表示树。下面是一个简单的示例:

```

class TreeNode:

def __init__(self, val):

self.val = val

self.left = None

self.right = None

# 创建一个二叉树

root = TreeNode(1)

root.left = TreeNode(2)

root.right = TreeNode(3)

root.left.left = TreeNode(4)

root.left.right = TreeNode(5)

# 遍历二叉树,打印每个节点的值

def traverse(node):

if node is None:

return

print(node.val)

traverse(node.left)

traverse(node.right)

traverse(root)

```

运行以上代码,输出结果为:

```

1

2

4

5

3

```

在上面的代码中,我们定义了一个 TreeNode 类,每个实例对象都表示一棵树的节点。每个节点包括一个值 `val`,以及左右子树的指针 `left` 和 `right`。

在创建二叉树的过程中,我们首先创建根节点,然后通过给根节点的 `left` 和 `right` 属性赋值来创建子树。例如,`root.left = TreeNode(2)` 表示创建了一个值为 2 的节点,并将其作为根节点的左子节点。

最后,我们定义了一个 traverse 函数,用来遍历二叉树并打印每个节点的值。在遍历过程中,我们首先打印当前节点的值,然后递归地遍历左子树和右子树,直到遍历完整棵树。


如果喜欢我的文章,可以关注我,我会分享更多python知识

最近发表
标签列表