力扣 面試題07. 重建二叉樹 遞歸分治/非遞歸版

https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/

思路：先序遍歷：根左右。中序遍歷：左根右。那麼可以由先序遍歷得到根節點的值，依據此值再找到根節點在中序遍歷的位置，那麼就可以分出左子樹和右子樹，分治下去即可。

/**
 * 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* dfs(int l1,int r1,int l2,int r2,vector<int>& preorder,vector<int>& inorder)
    {
        if(l1==r1)//只有一個節點時
        {
            TreeNode *node=new TreeNode(preorder[l1]);
            return node;
        }
        int rootval=preorder[l1];//根節點的值
        int i=l2;
        while(i<=r2&&inorder[i]!=rootval)//在中序遍歷中找到根節點
            ++i;
        int lnum=i-l2;//左子樹元素個數
        int rnum=r2-i;//右子樹元素個數
        TreeNode *root=new TreeNode(rootval);
        TreeNode *lc=NULL,*rc=NULL;
        if(lnum)
            lc=dfs(l1+1,l1+lnum,l2,i-1,preorder,inorder);
        if(rnum)
            rc=dfs(r1-rnum+1,r1,i+1,r2,preorder,inorder);
        root->left=lc;
        root->right=rc;
        return root;
    }

    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int siz=preorder.size();
        if(!siz)
            return NULL;
        return dfs(0,siz-1,0,siz-1,preorder,inorder);
    }
};

非遞歸版？感覺這個有點難理解，最好結合例子體會一下。

/**
 * 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* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int siz=preorder.size();
        if(!siz)
            return NULL;
        stack<TreeNode*> s;
        int j=0;
        TreeNode *node,*head=new TreeNode(preorder[0]);
        s.push(head);
        for(int i=1;i<siz;i++)
        {
            if(s.top()->val!=inorder[j])
            {
                node=new TreeNode(preorder[i]);
                s.top()->left=node;
                s.push(node);
            }
            else
            {
                while(!s.empty()&&s.top()->val==inorder[j])
                    node=s.top(),s.pop(),j++;
                TreeNode *tmp=new TreeNode(preorder[i]);
                node->right=tmp;
                s.push(tmp);
            }
        }
        return head;
    }
};