文章目錄

• • 1. 題目
  • 2. 棧解題

1. 題目

我們從二叉樹的根節點 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 之間。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/recover-a-tree-from-preorder-traversal
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 棧解題

• 用一個棧存儲pair<TreeNode*, depth>
• 將節點和其深度，不斷地壓棧
• 但是，當前節點的深度depth <= 棧頂的，說明需要往上面找父節點，所以彈棧
• 找到深度比我小的即是父節點，優先接在左節點

class Solution { public:TreeNode* recoverFromPreorder(string S) {int depth = 0, i = 0, val = 0;stack<pair<TreeNode*,int>> stk;TreeNode *root, *cur;val = getVal(S,i);root = new TreeNode(val);stk.push(make_pair(root,0));while(i < S.size()){depth = getDep(S,i);val = getVal(S,i);cur = new TreeNode(val);while(stk.top().second >= depth)stk.pop();if(stk.top().first->left)stk.top().first->right = cur;elsestk.top().first->left = cur;stk.push(make_pair(cur,depth));}return root;}int getVal(string &S, int &i)//獲取數字{int val = 0;while(i < S.size() && isdigit(S[i]))val = val*10+S[i++]-'0';return val;}int getDep(string &S, int &i)//獲取深度{int depth = 0;while(S[i] == '-'){depth++;i++;}return depth;} };

總結

以上是生活随笔為你收集整理的LeetCode 1028. 从先序遍历还原二叉树（栈）的全部內容，希望文章能夠幫你解決所遇到的問題。

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