1. 程式人生 > 其它 >0094-leetcode演算法實現之二叉樹中序遍歷-binary-tree-inorder-traversal-python&golang實現

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
}