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

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

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

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

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

思路:
先计算出每个节点是父亲(根)的第几代
然后再利用深度优先搜索,递归

代码如下:

/**
 * 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* build(queue<int>&gen,queue<int>& num,int depth){
     //递归查找
        if(gen.front()!=depth)  return NULL;
        TreeNode* root=new TreeNode(num.front());
        gen.pop();
        num.pop();
        root->left=build(gen,num,depth+1);
        root->right=build(gen,num,depth+1);
        return root;
    }
    TreeNode* recoverFromPreorder(string S) {
     
        int left=0;
        queue<int> gen,num;
        for(int i=0;i<S.size();i++){
     
            if(S[i]!='-'){
     
                gen.push(i-left);
                left=i;
                while(i<S.size()&&S[i]!='-') i++;
                num.push(stoi(S.substr(left,i-left)));
                left=i;
            }
        }
        return build(gen,num,0);//gen保存的是第几代,num保存的是节点数
    }
};

你可能感兴趣的:(leetcode,二叉树,leetcode,dfs,递归,c++)