我們在該專欄中記錄了我倆的刷題記錄。

我們更新的所有題目都在目錄中。

今天的題目是

力扣?leetcode-cn.com

題目

We run a preorder depth first search on the root of a binary tree.At each node in this traversal, we output D dashes (where D is the depth of this node), then we output the value of this node. (If the depth of a node is D, the depth of its immediate child is D+1. The depth of the root node is 0.)If a node has only one child, that child is guaranteed to be the left child.Given the output S of this traversal, recover the tree and return its root.我們從二叉樹的根節點 root 開始進行深度優先搜索。在遍歷中的每個節點處，我們輸出 D 條短劃線（其中 D 是該節點的深度），然后輸出該節點的值。（如果節點的深度為 D，則其直接子節點的深度為 D + 1。根節點的深度為 0）。如果節點只有一個子節點，那么保證該子節點為左子節點。給出遍歷輸出 S，還原樹并返回其根節點 root。Example 1

Input: "1-2--3--4-5--6--7"Output: [1,2,5,3,4,6,7]Example 2:

Input: "1-2--3---4-5--6---7"Output: [1,2,5,3,null,6,null,4,null,7]Example 3:

Input: "1-401--349---90--88"Output: [1,401,null,349,88,90]

思路 - DFS先序遍歷

這道題給的是先序遍歷的序列，那我們照著先序遍歷，使用一個輔助棧，把樹還原即可
在先序遍歷的過程中，如果不斷往下走，就不斷入棧，如果是往上走，就是出棧
注意，題目條件中說了，如果結點只有一個子節點，保證該子節點是左子節點。那么我們可以在建樹的時候，直接判斷當前結點的左子結點是否存在，不存在就加入到左子節點，存在就加入到右子節點
我們舉個例子直觀的說明一下

例如： 輸入："1-2--3--4-5" 輸出：[1,2,5,3,4,6,7]樹：stack = [] input = "1-2--3--4-5" 首先把1取出來，作為整個樹的root，并入棧樹：1 stack = [1] input = "-2--3--4-5" 把2取出來，以及2的depth為1 (depth為2前面的-的個數) 由于depth = len(stack)，說明還是往下走，取到棧頂1(不出棧)，且1的左子樹不存在，直接把2接到1的左子樹，同時2入棧樹：1/2 stack = [1,2] input = "--3--4-5" 把3取出來，以及3的depth為2 由于depth = len(stack)，說明還是往下走，取到棧頂2(不出棧)，且2的左子樹不存來，直接把3接到2的左子樹，同時3入棧樹：1/2/3 stack = [1,2,3] input = "--4-5" 把3取出來，以及3的depth為2 由于depth < len(stack)，說明往上走了，不斷出棧，使得depth = len(stack)，(這樣保證棧頂是3的父節點)，出棧后，stack=[1,2]，得到棧頂2 (不出棧)，2的左子樹存在，所以把4接到2的右子樹，同時4入棧樹：1/2/ 3 4 stack = [1,2,4] input = "-5" 把5取出來，以及5的depth為1 由于depth < len(stack)，說明繼續往上走，不斷出棧，使得depth=len(stack)，stack=[1]，得到棧頂1 (不出棧)，1的左子樹存在，所以把5接到1的右子樹，5入棧樹：1/ 2 5/ 3 4

所以整個過程中，我們需要維護一個棧，同時需要寫兩個輔助函數去找當前的結點值是多少，以及當前的結點深度是多少
我們來看代碼

# Definition for a binary tree node. # class TreeNode: # def __init__(self, x): # self.val = x # self.left = None # self.right = Noneclass Solution:def recoverFromPreorder(self, S: str) -> TreeNode:# 第一個輔助函數，給定起始點的index，找結點值def detectNum(start):ret = 0while start < len(S) and "0" <= S[start] <= "9":ret = ret * 10 + int(S[start])start += 1# 返回結點值，以及下一個起始點的indexreturn ret, start# 第二個輔助函數，給定起始點的index，找結點的深度def detectDepth(start):ret = 0while start < len(S) and S[start] == "-":ret += 1start += 1# 返回深度，以及下一個起始點的indexreturn ret, start if not S:return None# 找根結點的結點值val, idx = detectNum(0)# 結構化根結點root = TreeNode(val)# 根結點入棧stack = [root]while idx < len(S):# 檢測當前結點的深度depth, idx = detectDepth(idx)# 檢測當前結點的結點值val, idx = detectNum(idx)# 如果深度小于棧的深度，不斷出棧，使得二者深度相等# 這樣使得棧頂元素是當前結點的父結點while depth < len(stack):stack.pop()# 把父節點取出來parent = stack[-1]node = TreeNode(val)# 如果父節點無左子樹，則放到左子樹上# 反之，放到右子樹上if not parent.left:parent.left = nodeelse:parent.right = node# 當前點入棧stack.append(node)return root

總結

以上是生活随笔為你收集整理的js怎么取到遍历中的特定值_LeetCode 1028 hard 从先序遍历还原二叉树 Python解题记录...的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。