1008. 先序遍历构造二叉树/C++

返回与给定先序遍历 preorder 相匹配的二叉搜索树（binary search tree）的根结点。

(回想一下，二叉搜索树是二叉树的一种，其每个节点都满足以下规则，对于 node.left 的任何后代，值总 < node.val，而
node.right 的任何后代，值总 > node.val。此外，先序遍历首先显示节点的值，然后遍历 node.left，接着遍历
node.right。）

示例：

输入：[8,5,1,7,10,12] 输出：[8,5,10,1,7,null,12]

提示：

1 <= preorder.length <= 100
2 先序 preorder 中的值是不同的。

整个过程就是操作3个节点
node：依据先序遍历生成的节点
pre：node插入的节点
top：栈顶节点，也是与node的值相比较的节点。

• 依据先序遍历构造一个待插入节点node
• 比较栈顶节点top的值与node的值
  • 如果node的值小于top的值，直接将node插入pre的左边。并且将node入栈，将node赋值给pre。
  • 如果node的值大于top的值，将栈顶元素赋值给pre，出栈，然后比较新top与node的值。
    • 直到栈为空，或者node的值小于top的值。将node插入pre的右边。

第二个分支比较绕，举个例子，要插入例题中的7时，要与8相比较，然后插入到5的右边。此时，node是7，pre是5，top是8.

/**
 * 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* bstFromPreorder(vector<int>& preorder) {
        stack<TreeNode*> stk;
        TreeNode* root = new TreeNode(preorder[0]);
        stk.push(root);
        TreeNode* pre = root;
        for(int i=1;i<preorder.size();++i){
            TreeNode* node = new TreeNode(preorder[i]);
            TreeNode* top = stk.top();
            
            if(node->val < top->val){
                pre->left = node;
                stk.push(node);
                pre=node;
            }
            else{
                pre=stk.top();
                stk.pop();
                while(!stk.empty()){
                    TreeNode* top = stk.top();
                    if(top->val > node->val){
                        break;
                    }else{
                        pre=top;
                        stk.pop();
                    }
                }
                pre->right=node;
                stk.push(node);
                pre=node;
            }
        }
        return root;
    }
};