0094-leetcode演算法實現之二叉樹中序遍歷-binary-tree-inorder-traversal-python&golang實現
阿新 • • 發佈:2021-11-12
給定一個二叉樹的根節點 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 }