【先序遍历 深度优先搜索】1028. 从先序遍历还原二叉树

本文涉及知识点

先序遍历
深度优先搜索汇总

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

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

深度优先搜索

读取第一个数作为root，然后深度优先。
DFS(Vector<TreeNode*> pars,const string& s ,int iPos )
cnt 等于-的数量。
num的后面的值
$$\{\begin{array}{llc}pars出栈，直到cnt等于pars.size()pars[cnt-1]->right是当前节点& & cnt<pars.size()\\ pars[cnt-1]->left是当前节点& & cnt==pars.size()\end{array}$$

代码

class Solution {
public:
	TreeNode* recoverFromPreorder(string traversal) {
		int num = 0;
		int pos = 0;
		for (; isdigit(traversal[pos]); pos++) {
			num = num * 10 + (traversal[pos] - '0');
		}
		TreeNode* root = new TreeNode(num);
		vector< TreeNode*> pars;
		pars.emplace_back(root);
		DFS(pars, traversal, pos);
        return root;
	}
	void DFS(vector< TreeNode*>& pars, const string traversal, int pos) {
        if (pos >= traversal.length()) {return ;}	
		int cnt = 0;
		for (; '-' == traversal[pos]; pos++,cnt++);
		int num = 0;
		for (; (pos < traversal.length()) &&  isdigit(traversal[pos]); pos++) {
			num = num * 10 + (traversal[pos] - '0');
		}
		TreeNode* node = new TreeNode(num);
		if (pars.size() == cnt) {
			pars.back()->left = node;
		}
		else {
			while (pars.size() > cnt) {
				pars.pop_back();
			}
			pars.back()->right = node;
		}
		pars.push_back(node);
        DFS(pars, traversal, pos);			
	}
};

扩展阅读

视频课程

先学简单的课程，请移步CSDN学院，听白银讲师（也就是鄙人）的讲解。
https://edu.csdn.net/course/detail/38771

如何你想快速形成战斗了，为老板分忧，请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176

相关推荐

我想对大家说的话
《喜缺全书算法册》以原理、正确性证明、总结为主。
按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
有效学习：明确的目标 及时的反馈 拉伸区（难度合适） 专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛

测试环境

操作系统：win7 开发环境： VS2019 C++17
或者 操作系统：win10 开发环境： VS2022 C++17
如无特殊说明，本算法用**C++**实现。