LeetCode 1028. 从先序遍历还原二叉树 | Python

1028. 从先序遍历还原二叉树

---

题目来源：力扣（LeetCode）https://leetcode-cn.com/problems/recover-a-tree-from-preorder-traversal

题目

---

我们从二叉树的根节点 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 之间。

解题思路

---

思路：栈 + 迭代

根据题意，结合示例和图例，我们可以得到如下信息：

• 遍历字符串时，读取 - 字符，直至遇到非 - 字符。此时我们可以通过 - 的个数判断当前节点的深度。（也印证题目中所说【在遍历中的每个节点处，我们输出 D 条短划线（其中 D 是该节点的深度）】）；
• 读取数字（