给定一个二叉树的dfs遍历结果(NULL记为*)，重构二叉树，返回头节点

给定一个二叉树的dfs遍历结果(NULL记为*)，重构二叉树，返回头节点。

思路：第一遍先把*也插入到树中，第二遍把*改成NULL。

如果直接把*记录为NULL，那再来一个节点就不知道，到底是*赋值的NULL，还是本身自带的NULL。

#include <bits/stdc++.h> 
using namespace std; 

struct TreeNode{
    char val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(char nval){
        val = nval;
        left = NULL;
        right = NULL;
    }    
}; 

TreeNode* construct(string str){
    if(str.empty()) return NULL;
    
    stack<TreeNode*> temp;
    TreeNode* root = new TreeNode(str[0]);    
    temp.push(root);
    
    for(int i=1; i<str.size(); i++){
        if( str[i]>='a' && str[i]<='z' ){
            TreeNode* newNode = new TreeNode(str[i]);
            while(temp.top()->left && temp.top()->right ) temp.pop();
            
            if(temp.top()->left ==NULL) temp.top()->left = newNode;
            else if(temp.top()->right==NULL ) {
                temp.top()->right= newNode;
            }            
            temp.push(newNode);                        
        }    
        else if(str[i]=='*'){
            TreeNode* newNode = new TreeNode(str[i]);
            if(temp.top()->left ==NULL) temp.top()->left = newNode;
            else if( temp.top()->right==NULL ){
                temp.top()->right = newNode;
                temp.pop();
            } 
        }
        
    } 
    
    //按层遍历,将*改为NULL   
    queue<TreeNode*> que;
    TreeNode* cur = NULL;
    que.push(root);    
    while( !que.empty() ){
        int size = que.size();
        while(size--){
            cur = que.front();
            que.pop();
            if(cur->left && cur->left->val=='*'){
                cur->left = NULL;
            }    
            if(cur->left && cur->left->val!='*'){
                que.push(cur->left);
            }
            
            if(cur->right && cur->right->val == '*'){
                cur->right = NULL;
            }
            if(cur->right && cur->right->val !='*'){
                que.push(cur->right);
            }            
        }         
    }
        
    return root;
    
} 

void search(TreeNode* root, string& res){
    if(root==NULL){
        res+='*';
        return;
    } 
//    res += root->val;
    search(root->left, res);    
    res += root->val;
    search(root->right, res);
}


int main(){
//    string str = "abdm***n**ew**cf**g**";
    string str = "ab*gf***c*de**f**"; 
    TreeNode* root = construct(str);
        
    string out="";
    search(root, out);
    
    cout<<out<<endl;
    
}