1. 题目
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
//前序遍历 preorder = [3,9,20,15,7]
//中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
2. 分析
这个题目是以前在本科的数据结构考试中经常出现的一类问题。前序遍历顺序是遍历根节点,左子树,右子树,而中序遍历则是左子树,根节点,右子树,因此这类题目的解题思路是根据前序遍历的第一个元素确定根节点,然后在中顺遍历中找到根节点的位置。在中序遍历的左侧是左子树,右侧是右子树。如上面的例子,首先我们根据前序的第一个节点确定3是根节点,那么在中序遍历结果中找到3,那么中序遍历结果中左侧的序列【9】则是3为根节点的左子树的中序结果,而右侧的序列【15,20,7】则是右子树的中序结果。
确定了左右子树,继续在左子树的中序遍历结果中找到出现在先序遍历结果的元素,因为在先序遍历结果首先出现的一定是子树的根节点。如本题,左子树的中序遍历结果为【9】,只有一个元素,那么一定是9先出现在先序的结果中,因此左子树根节点为9。右子树的中序遍历结果为【15,20,7】,那么首先出现在先序遍历结果【3,9,20,15,7】的元素是20,那么20是右子树的根节点。
因为左子树根节点9在其子树对应的中序结果【9】中没有左侧和右侧的序列,那么9则是一个叶子节点。而右子树根节点20在其对应子树的中序结果【15,20,7】中存在左侧序列【15】和右侧序列【7】,那么【15】对应的则是以20为根节点的左子树的中序结果,而【7】则是以20为根节点的右子树的中序结果。循环递归上面的过程构造子树。
上面的过程是我在数据结构考试中,笔试经常使用的解题思路,反应到程序中需要解决两个重要的问题:
1. 先序遍历结果的第一个元素(根节点)在中序遍历结果中的位置如何确定?
2. 左子树中序遍历子序列的根节点,即左子树的根节点如何确定?同样,右子树中序遍历子序列的根节点,即右子树的根节点如何确定?
3. C++程序一(版本1:leetcode提交超出时间限制)
这份代码能够通过一些简单的树的case,但是如果树的数量级太大,在leetcode的203个测试中,通过了202个,最后一个测试显示超时,因为这个算法的时间复杂度是3次方级!!!。
虽然这份程序没有通过所有测试,但我还是在这里给出,主要原因是我想通过在这份程序的思路上纠正可以优化的部分。
也许有的读者能够发现,前面的分析中,我始终只是对中序结果进行划分,而始终在原始的preorder序列中确定位置。
反观我们的代码中,首先,在中序遍历结果中确定根节点位置这个过程需要保留,而这个过程的时间复杂度是线性的。程序中主要的时间消耗在于从子树对应的中序遍历结果中确定根节点这个过程。因为需要一层循环逐个遍历中序遍历子序列的元素,然后在先序结果中逐个确定其位置,确定位置的过程也需要循环,则是两层循环,体现在下面的代码段中。
for(int i=0;i<left.size();i++){
int index = 0;
while(left[i]!=preorder[index]) index++;
if(index<minLeft){
leftVal = preorder[index];
minLeft = index;
}
}
再加上递归的过程,总的时间复杂度是,其中logn是递归的次数,即树高,n^2则是上面给出的两层循环