中序遍历代码
- 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
- 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
- 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。
中序遍历代码
中序遍历是二叉树的一种遍历方式,它按照“左子树-根节点-右子树”的顺序访问二叉树的所有节点。
在编写中序遍历代码时,我们可以使用递归或迭代的方式来实现。
递归实现中序遍历
递归是一种简洁而直观的方法来实现中序遍历。
下面是递归实现中序遍历的代码:
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def inorderTraversal(root):
if root is None:
return []
result = []
result.extend(inorderTraversal(root.left))
result.append(root.val)
result.extend(inorderTraversal(root.right))
return result
在这段代码中,我们定义了一个TreeNode类来表示二叉树的节点。
每个节点包含
一个值val、左子节点left和右子节点right。
函数inorderTraversal()接受一个二叉树的根节点作为参数,并返回一个列表,其
中包含了按照中序遍历顺序访问得到的所有节点值。
该函数首先进行终止条件判断:如果当前节点为空,则直接返回空列表。
否则,我们先通过递归调用处理左子树,将结果添加到结果列表中。
然后将当前节点的值添加到结果列表中。
最后,再通过递归调用处理右子树,并将结果添加到结果列表中。
最终返回结果列表。
迭代实现中序遍历
除了使用递归,我们还可以使用迭代的方式来实现中序遍历。
迭代方式通常借助栈来辅助实现。
下面是迭代实现中序遍历的代码:
def inorderTraversal(root):
if root is None:
return []
result = []
stack = []
node = root
while node or stack:
while node:
stack.append(node)
node = node.left
node = stack.pop()
result.append(node.val)
node = node.right
return result
在这段代码中,我们同样定义了一个TreeNode类来表示二叉树的节点。
函数inorderTraversal()接受一个二叉树的根节点作为参数,并返回一个列表,其中包含了按照中序遍历顺序访问得到的所有节点值。
我们首先进行终止条件判断:如果当前节点为空,则直接返回空列表。
然后,我们初始化一个空列表result来保存最终结果。
同时,我们还需要使用一个栈stack来辅助迭代过程。
初始时,将根节点入栈。
接下来,我们进入循环,当循环条件满足时执行以下操作:
•在内层循环中,将当前节点的左子节点依次入栈,直到当前节点为空。
•弹出栈顶元素,将其值添加到结果列表中。
•将当前节点更新为栈顶元素的右子节点。
通过以上操作,我们可以实现按照中序遍历顺序访问所有节点,并将其值添加到结果列表中。
最后返回结果列表。
总结
中序遍历是一种常用的二叉树遍历方式,它按照“左子树-根节点-右子树”的顺序访问二叉树的所有节点。
我们可以使用递归或迭代的方式来实现中序遍历。
递归方式简洁而直观,但可能存在函数调用的开销。
迭代方式利用栈的辅助结构,能够在不使用递归调用的情况下完成遍历过程。
无论是递归还是迭代实现中序遍历,都需要注意处理空节点的情况,并在合适的位置将节点值添加到结果列表中。
通过理解和掌握中序遍历代码的实现方式,我们可以更好地理解和应用二叉树相关算法和数据结构。