思路:
1.这道题还是借鉴了其他大佬的思路,仔细分析一下,前序遍历的第一个值是根节点值(假设为val),那么在中序遍历中val之前的都是根节点的左子树,val值之后的都是右子树。同理递归可还原二叉树。
2.容易出错的地方:
a.递归的子函数最好传入前序和中序的引用,然后每次更新前序和中序遍历的起始点和终止点,这样比较不容易出错。
b.起始点和终止点的计算需要细心~尤其是得到middle之后,一定要和ino_s相减才能得到中间间隔的值得个数。
c.顺便总结一下,前序+中序或者后序+中序都可以得到二叉树,但是前序+后序得不到二叉树。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode *BuildTreePI(vector<int> &preorder, vector<int>&inorder,int pre_start,int pre_end,int ino_start,int ino_end){
printf("p_s = %d\np_e = %d\n",pre_start,pre_end);
if(pre_start>pre_end) return NULL;
int middle =ino_start; //这里很重要!!!middle从一开始就赋值为中序的开头,并且作为后面的自加变量!!
int com = preorder[pre_start];
printf("com = %d\n",com);
printf("val = %d\n",middle);
for(; middle<ino_end;middle++){ //这里要用middle作为自增变量
if(inorder[middle]==com){ //以及上面的for循环里,middle<ino_end,而不是小于等于。因为最大值只能是middle(也就是ino_start)和ino_end相等
break; //最后还有一次自加的!
}
}
/*
for(int i = ino_start; i<ino_end;middle++){ //不能这么写,因为如果i等于ino_end-1时,不会进入if判断 if(inorder[i]==com){ //也就是middle始终等于初始值ino_start middle = i; //而真正的i已经在经过最后一次循环后自加一次,等于ino_end了
printf("middle_out = %d\n",middle); TreeNode* p = new TreeNode(com); p->left = BuildTreePI(preorder,inorder,pre_start+1,middle-ino_start+pre_start,ino_start,middle-1); p->right = BuildTreePI(preorder,inorder,pre_end-ino_end+middle+1,pre_end,middle+1,ino_end); return p; } TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) { return BuildTreePI(preorder,inorder,0,preorder.size()-1,0,inorder.size()-1); } };break; } }*/