【問題描述】[困難]

我們從二叉樹的根節點 root 開始進行深度優先搜索。

在遍歷中的每個節點處，我們輸出 D 條短劃線（其中 D 是該節點的深度），然后輸出該節點的值。（如果節點的深度為 D，則其直接子節點的深度為 D + 1。根節點的深度為 0）。

如果節點只有一個子節點，那么保證該子節點為左子節點。

給出遍歷輸出 S，還原樹并返回其根節點 root。

示例 1：

輸入：“1-2–3--4-5–6--7”
輸出：[1,2,5,3,4,6,7]
示例 2：

輸入：“1-2–3—4-5–6—7”
輸出：[1,2,5,3,null,6,null,4,null,7]
示例 3：

輸入：“1-401–349—90–88”
輸出：[1,401,null,349,88,90]

提示：

原始樹中的節點數介于 1 和 1000 之間。
每個節點的值介于 1 和 10 ^ 9 之間。

【解答思路】

迭代方法（模擬遞歸）

時間復雜度：O(N) 空間復雜度：O(N)

public TreeNode recoverFromPreorder(String S) {Deque<TreeNode> path = new LinkedList<TreeNode>();int pos = 0;while(pos < S.length()){int level = 0;// 數-while (S.charAt(pos) == '-') {++level;++pos;}//算數 int value =0 ;while(pos < S.length() && Character.isDigit(S.charAt(pos)) ){value = value*10 +(S.charAt(pos)-'0');++pos;}TreeNode node =new TreeNode(value);//左節點if(level == path.size()){if(!path.isEmpty()){path.peek().left =node; }}else{//找右節點while(level!=path.size()){path.pop();}path.peek().right =node; }path.push(node);}while(path.size()>1){path.pop();}return path.peek();}

【總結】

1.題解里寫了「迭代」而不是「遞歸」的方法。其實可以發現，如果我們用「遞歸」來實現本題的話，必須也要用到一些全局變量來維護例如「當前遍歷到字符串的哪個位置」「下一個節點的深度」這類的信息。這樣給我的感覺就像是「在遞歸的模板上套了一些遞歸沒有辦法做的事情」。既然如此的話，我們不如就將遞歸直接改寫成迭代的形式，每次添加一個新節點的時候，就將一些節點從我們自己維護的棧中彈出，將「用棧來遍歷樹」和「用全局變量實時維護信息」這兩者統一起來。

2.以鏈表組的頭一個作為答案的思路有點妙

Deque path = new LinkedList();

3. 棧

pop() 出
peek() 棧頂
push() 入

轉載鏈接：https://leetcode-cn.com/problems/recover-a-tree-from-preorder-traversal/solution/cong-xian-xu-bian-li-huan-yuan-er-cha-shu-by-leetc/

參考鏈接：https://leetcode-cn.com/problems/recover-a-tree-from-preorder-traversal/solution/shou-hui-tu-jie-fei-di-gui-fa-zhong-gou-chu-er-cha/

總結

以上是生活随笔為你收集整理的[leetcode][JAVA]面试题第[1028]题[迭代]的全部內容，希望文章能夠幫你解決所遇到的問題。

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