【leetcode】105.（Medium）Construct Binary Tree from Preorder and Inorder Traveral

最新推荐文章于 2021-12-13 22:59:20 发布

AXIMI

最新推荐文章于 2021-12-13 22:59:20 发布

阅读量148

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/AXIMI/article/details/85276585

版权

leetcode 专栏收录该内容

257 篇文章 2 订阅

订阅专栏

解题思路：
递归
先序遍历的第一个结点是根节点
根据根节点和中序遍历的结果可以找到左子树的结点和右子树的结点。
继续递归，将左子树的结点和右子树的结点构造成树，最后一层层创接起来。

时间复杂度：O(nlogn) 递归每层的耗时为O(n)，共logn层
空间复杂度：O(nlogn) 递归每层所占空间为O(n)，共logn层

如果将函数写成指针的形式（即不新建左子树和右子树的数组，直接在原数组中指明左子树和右子树的范围），时间复杂度可提升到O(logn)

提交代码：

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder.length==0)	return null;
        if(preorder.length==1) {
        	TreeNode node=new TreeNode(preorder[0]);
        	return node;
        }
        TreeNode root=new TreeNode(preorder[0]);
        
        int i,j,leftSize=0;
        for(i=0;i<inorder.length&&inorder[i]!=root.val;i++)
        	leftSize++;
        
        int[] leftInOrder=new int[leftSize];
        int[] leftPreOrder=new int[leftSize];
        int[] rightInOrder=new int[inorder.length-leftSize-1];
        int[] rightPreOrder=new int[inorder.length-leftSize-1];
        
        for(i=0;i<leftSize;i++)
        	leftInOrder[i]=inorder[i];
        for(i=leftSize+1,j=0;i<inorder.length;i++,j++)
        	rightInOrder[j]=inorder[i];
        
        for(i=0;i<leftSize;i++)
        	leftPreOrder[i]=preorder[i+1];
        for(i=leftSize+1,j=0;i<preorder.length;i++,j++)
        	rightPreOrder[j]=preorder[i];
        
    		root.left=buildTree(leftPreOrder,leftInOrder);
    		root.right=buildTree(rightPreOrder,rightInOrder);
    		return root;
    	}
}