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

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

我们从二叉树的根节点 root 开始进行深度优先搜索。

在遍历中的每个节点处，我们输出 D 条短划线（其中 D 是该节点的深度），然后输出该节点的值。（如果节点的深度为 D，则其直接子节点的深度为
D + 1。根节点的深度为 0）。

如果节点只有一个子节点，那么保证该子节点为左子节点。

给出遍历输出 S，还原树并返回其根节点 root。

思路

可以从字符串S中读取节点的值和深度信息。

• 读取字符的’-’，直到遇到数字，通过’-'的个数来知道节点的深度信息。
• 读取数字字符直到遇到非数字字符，获得节点的值。

假定当前节点为P，上一节点为S，只有两种情况：

• P为S的左子节点，P的深度比S的深度大1。
• P是S的路径上的某一节点的右节点（不包括S），因为题目规定，节点只有一个子节点时，保证该子节点为左子节点，由于先序遍历的性质，左子节点肯定比右子节点先访问。

因而可以使用一个栈来保存深度信息，当获得节点P的信息时，具体做法：

• 当P的深度等于栈顶节点的深度 + 1时，设置P为栈顶节点的左子节点。
• 当P的深度不等于栈顶节点的深度 + 1，将栈顶节点出栈，直到找到P的深度等于栈顶节点的深度 + 1，设置P为栈顶节点的右子节点。
• 最后将P节点进栈。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* recoverFromPreorder(string S) {
        int n = S.size();
        stack<TreeNode*> s;
        int pos = 0;
        while(pos < n) {
            int level = 0;
            while(S[pos] == '-') {
                pos++;
                level++;
            }
            int num = 0;
            while(pos < n && isdigit(S[pos])) {
                num = num * 10 + (S[pos] - '0');
                pos++;
            }
            TreeNode* node = new TreeNode(num);
            //判断栈顶节点和当前节点的深度信息
            //相等即设置为栈顶节点的左子节点
            if(level == s.size()) {
                if(!s.empty()) {
                    s.top() -> left = node;
                }
            }
            //不相等出栈直至相等，设置为栈顶节点的右子节点
            else {
                while(level != s.size()) {
                    s.pop();
                }
                s.top() -> right = node;
            }
            s.push(node);
        }
        //栈底的节点必为根节点
        while(s.size() > 1) {
            s.pop();
        }
        return s.top();
    }
};