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

题目链接

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

思路

先序遍历的顺序是根左右，中序遍历的顺序是左根右，所以在 preorder 数组中的首元素一定是当前树的树根，再从 inorder 数组中找到这个元素的下标 index，这样 index 的左边是左子树，右边是右子树。之后再分别递归建立左右子树即可。

递归的终止条件是传入的开始和结束下标不合法。

注意： preorder 数组中左右子树的开始和结束下标可以通过在 inorder 数组中分别计算左右子树元素个数推算。

代码

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return buildTree(preorder, 0, preorder.length - 1,
                inorder, 0, inorder.length - 1);
    }

    private TreeNode buildTree(int[] preorder, int preStart, int preEnd, int[] inorder, int inStart, int inEnd){
        if(preStart > preEnd || inStart > inEnd){
            return null;
        }

        int current = preorder[preStart];
        TreeNode root = new TreeNode(current);
        // find current value index in inorder
        int index = inStart;
        while(index <= inEnd){
            if(inorder[index] == current){
                break;
            }
            index++;
        }

        root.left = buildTree(preorder, preStart + 1, preStart + index - inStart,
                inorder, inStart, index - 1);
        root.right = buildTree(preorder, preStart + index - inStart + 1, preEnd,
                inorder, index + 1, inEnd);

        return root;
    }
}