0094-leetcode算法实现之二叉树中序遍历-binary-tree-inorder-traversal-python&golang实现

给定一个二叉树的根节点 root ，返回它的 中序 遍历。

示例 1：

输入：root = [1,null,2,3]
输出：[1,3,2]
示例 2：

输入：root = []
输出：[]
示例 3：

输入：root = [1]
输出：[1]
示例 4：

输入：root = [1,2]
输出：[2,1]
示例 5：

输入：root = [1,null,2]
输出：[1,2]

提示：

树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-inorder-traversal

python

# 0094.二叉树中序遍历
# 递归 & 迭代

class Solution:
    def inOrderRecur(self, head: TreeNode) -> int:
        """
        递归遍历，LNR, 左根右
        :param head:
        :return:
        """
        def traversal(head):
            # 递归终止条件
            if head == None:
                return
            traversal(head.left)
            print(head.val + " ")
            res.append(head.val)
            traversal(head.right)
        res = []
        traversal(head)
        return res

    def inorderItration(self, head: TreeNode):
        """
        迭代遍历，LNR, 左根右
        :param head:
        :return:
        """
        if head == None:
            return

        cur = head
        stack = []
        res = []
        while cur or stack:
            # 先迭代访问最底层的左子树节点
            if cur:
                stack.append(cur)
                cur = cur.left
            # 到达最左节点后处理栈顶节点
            else:
                cur = stack.pop()
                res.append(cur.val)
                # 取栈顶元素的右节点
                cur = cur.right
        return res

golang

package main

import "container/list"

// 二叉树的中序遍历 -> 递归 && 迭代
// 递归遍历
func InOrderTraversal(root *TreeNode) []int {
	// 递归遍历， LNR, 左根右
	var res = []int{}
	var inorder func(node *TreeNode)
	inorder = func(node *TreeNode) {
		if node == nil {
			return
		}
		inorder(node.Left)
		res = append(res, node.Val)
		inorder(node.Right)
	}
	inorder(root)
	return res
}

// 迭代遍历 LNR 左中右 左根右
func InOrder(root *TreeNode) []int {
	var res = []int{}
	if root == nil {
		return nil
	}
	stack := list.New() // 创建链表容器
	node := root

	// 1.先将所有的左节点找到，压入栈中
	for node != nil {
		stack.PushBack(node)
		node = node.Left
	}
	// 2.对栈中的每个节点先弹出加入到res中，再找到该节点的右节点的所有左节点加入栈中
	for stack.Len() > 0 {
		e := stack.Back()
		node := e.Value.(*TreeNode)
		stack.Remove(e)
		// 找到该节点的右节点，再搜索其所有的左节点加入栈中
		res = append(res, node.Val)
		node = node.Right
		for node != nil {
			stack.PushBack(node)
			node = node.Left
		}
	}
	return res
}