文章目录
- 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;}
};