一般笔试题都会让你根据二叉树的前序遍历和后续遍历求出二叉树。
当然,手写的时候还是很简单的,用代码,稍微有点麻烦。在这里,我们还是采用递归的思想,去求解。
假设给定一个前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
首先,我们要在前序遍历中寻找根节点。前序遍历,访问顺序是根,左,右,首先访问的根节点。
那么{1,2,4,7,3,5,6,8}中,1是根节点,那么其余的节点要么在左子树,要么是右子树。
那么如何确定左子树和右子树呢?
这就需要查看中序遍历了。中序遍历的顺序是:左,根,右。
那么只要在中序遍历中,寻找出根节点,就可以确定,左子树,右子树。
例如 1是根节点,那么4,7,2 这些节点肯定在1的左边。5,3,8,6根节点1的右边。
那么,这时候可以去确定左子树的先序遍历,中序遍历。右子树的前序遍历,中序遍历。
实质上,取出前序遍历,就是要根据左子树的长度,右子树的长度。
struct TreeNode* recur(vector<int> pre,vector<int> in,int istart,int iend,int pstart,int pend) {
<span style="white-space:pre"> <span style="white-space:pre"> </span>
<span style="white-space:pre"> </span>/*istart,iend:中序遍历的开始和结束
pstart,pend:前序遍历的开始和结束
<span style="white-space:pre"> </span>*/</span>
<span style="white-space:pre"> </span>/*构建二叉树*/
TreeNode * head= new TreeNode(0) ;
head->val=pre[pstart];
head->left=NULL;
<span style="white-space:pre"> </span>head->right=NULL;
<span style="white-space:pre"> </span>int i=istart;
<span style="white-space:pre"> </span>if(istart==iend&&pstart==pend){
<span style="white-space:pre"> </span>return head;
<span style="white-space:pre"> </span>}
if(istart<=iend&&pstart<=pend)
{
for(;in[i]!=pre[pstart]&&(i<=iend);i++);//从中序遍历中寻找根节点的位置
if(i-1>=istart&&(i-istart+pstart>=pstart+1))
head->left = recur(pre,in, istart, i,pstart+1, i-istart+pstart);
if(iend>=i+1&&(pend>=i-istart+pstart+1))
head->right = recur(pre,in, i+1, iend, i-istart+pstart+1, pend);
}
return head;
}
struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {
TreeNode * head = NULL;
<span style="white-space:pre"> </span>if(in.size()==0)
<span style="white-space:pre"> </span>return NULL;
<span style="white-space:pre"> </span>int end=in.size()-1;
<span style="white-space:pre"> </span>return recur(pre,in, 0, end, 0, end);
}