一、题目:剑指 Offer 07. 重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
限制:
0 <= 节点个数 <= 5000
注意:本题与主站105. 从前序与中序遍历序列构造二叉树重复,刷一个就是刷两个,血赚!
二、思路和代码:
核心知识点:
1、前序遍历:根->左->右
2、中序遍历:左->根->右
3、前序遍历后的序列中的第一个元素就是根节点;通过在中序遍历的数组中找到这个根可以确定左右子树。
上一句话在左右子树的元素子序列中依然适用。
这不就可以递龟了吗?
递龟就是函数自己调用自己。
/**
* 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* buildTree(vector<int>& preorder, vector<int>& inorder) {
if(preorder.size()==0)return NULL;
int temp =0;
TreeNode* root = new TreeNode(preorder[temp]);
//用temp找到根节点
while(inorder[temp] != root->val)temp++;
//用temp划分前序和中序的左子树来
vector<int>p_L(preorder.begin()+1,preorder.begin()+temp+1);
vector<int>i_L(inorder.begin(),inorder.begin()+temp);
root->left = buildTree(p_L,i_L);
//用temp划分前序和中序的右子树来
vector<int>p_R(preorder.begin()+temp+1,preorder.end());
vector<int>i_R(inorder.begin()+temp+1,inorder.end());
root->right = buildTree(p_R,i_R);
return root;
}
};
PS:中序后续也是可以建树的。总而言之重建树一定要有中序遍历。
怕什么真理无穷,进一步有进一步的欢喜!