LeetCode 105. 从前序与中序遍历序列构造二叉树

/**
     * leetCode 105. 从前序与中序遍历序列构造二叉树
     * 参考解析：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by--22/
     * @param preorder 前序
     * @param inorder  中序
     * @return  二叉树的根节点
     *  思路：1：先构建二叉树的根节点，根节点就是前序数组的第一个元素，
     *        2：按照前序根节点--》左子树--》右子数的特点，遍历前序数组构建左子树中的左侧节点
     *        3：如果前序的节点遍历到中序的第一个节点，说明左子树左侧节点的遍历已经完成，需要反向遍历栈中存在的节点，结合中序遍历补全左子树的右侧节点
     *        4.同理，通过这种方式，构建右子树
     */
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder.length == 0) {
            return null;
        }
        int pre_index = 0;
        int in_index = 0;
        //前序的第一个元素肯定是根节点，没跑了
        TreeNode root = new TreeNode(preorder[0]);
        pre_index++;
        Stack<TreeNode> stack = new Stack<>();
        TreeNode currNode = root;
        stack.push(currNode);
        while (pre_index < preorder.length) {
            //根据中序判断是左子树还是右子数
            if(currNode.val==inorder[in_index]){
                while (!stack.isEmpty() && stack.peek().val==inorder[in_index]) {
                    currNode = stack.peek();
                    stack.pop();
                    in_index++;
                }
                //右子树
                currNode.right = new TreeNode(preorder[pre_index]);
                currNode = currNode.right;
                stack.push(currNode);
                pre_index++;
            }else{
                //左子树
                currNode.left =  new TreeNode(preorder[pre_index]);
                currNode = currNode.left;
                stack.push(currNode);
                pre_index++;
            }
        }
        return root;
    }


    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }