（DFS）1028. 从先序遍历还原二叉树_dfs 序中还原树的本来面貌-优快云博客

本文介绍了一种从深度优先遍历序列中恢复二叉树结构的方法，通过解析字符串来确定节点值和深度，进而构建二叉树。文章包含示例和代码实现，展示了如何使用深度优先搜索算法解决这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我们从二叉树的根节点 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解：将每个结点与其所在层数记录下来，通过dfs构建二叉树即可。

/**
 * 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:
    int change(string str){
        int ans=0, len=str.length();
        for(int i=0;i<len;i++) ans = ans * 10 + (str[i]-'0');
        return ans;
    }
    
    TreeNode* dfs(list<int>& nums, list<int>& dep, int index){
        if(dep.front() != index) return NULL;  // 当前层数与递归层数不匹配，返回空子树
        TreeNode* rt = new TreeNode(nums.front());
        nums.pop_front(); dep.pop_front();
        rt->left = dfs(nums, dep, index + 1);
        rt->right = dfs(nums,dep, index + 1);
        return rt;
    }
    
    TreeNode* recoverFromPreorder(string str) {
        int len = str.length(), i = 0, pos = 0;
        if(!len) return NULL;
        list<int>nums,dep;
        while(i<len){
            int j=i;
            while(j<len && str[j]!='-') j++;
            nums.push_back(change(str.substr(i,j-i)));
            dep.push_back(i-pos);
            if(j >= len) break;
            pos = j;
            i = j;
            while(i<len && str[i]=='-') i++;
        }
         
        return dfs(nums, dep, 0);
    }
};