题目难度: Medium
原题描述:
Given preorder and inorder traversal of a tree, construct the binary tree.
Note:
You may assume that duplicates do not exist in the tree.
题目大意:
给出一棵二叉树的前序遍历和中序遍历的结果,让你建立这棵二叉树。
解题思路:
这题主要考察前序遍历和中序遍历的过程和通过递归来建树。要注意到前序遍历一棵二叉树的第一个结点是这棵树的根节点,接着在中序遍历中找到这个根节点,它的左边部分就是左子树,右边部分是右子树。根据中序遍历中左子树结点的数量可以在前序遍历中划分出左子树和右子树。这样,前序遍历和中序遍历都划分出了根节点、左子树和右子树,接下来创建根节点,分别递归地创建左子树和右子树。
时间复杂度分析:
平均情况下,二叉树左右子树的高度差不多,因此可以假设根结点处在中序遍历的中点,而每次在中序遍历中寻找根结点的时间复杂度为O(n),因此可以得到如下递归式:f(n) = 2f(n/2) + n。解得f(n) = O(n*log(n))。
以下是代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
struct TreeNode* createTree(int* preorder, int x1, int y1, int* inorder, int x2, int y2)
{
if(x1>y1 || x2>y2)
return NULL;
int rootIndex;
for(int i=x2 ; i<=y2 ; ++i){
if(inorder[i] == preorder[x1]){
rootIndex = i;
break;
}
}
int leftTreeNum = rootIndex - x2;
struct TreeNode * root = (struct TreeNode *)malloc(sizeof(struct TreeNode));
root->val = preorder[x1];
root->left = createTree(preorder, x1+1, x1+leftTreeNum, inorder, x2, rootIndex-1);
root->right = createTree(preorder, x1+leftTreeNum+1, y1, inorder, rootIndex+1, y2);
return root;
}
struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize)
{
return createTree(preorder, 0, preorderSize-1, inorder, 0, inorderSize-1);
}