根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/
9 20
/
15 7
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* 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;
TreeNode *root=new TreeNode(preorder[0]);
if(root==NULL)
return root;
vector<int>pre_left,pre_right,vin_left,vin_right;
int gen;
int len=preorder.size();
for(int i=0;i<len;i++)//寻找根节点在中序排序中的位置
{
if(inorder[i]==preorder[0])
{
gen=i;
break;
}
}
for(int i=0;i<gen;i++)//将左子树存入左动态数组中
{
pre_left.push_back(preorder[i+1]);
vin_left.push_back(inorder[i]);
}
for(int i=gen+1;i<len;i++)//将右子树存入动态数组中
{
pre_right.push_back(preorder[i]);
vin_right.push_back(inorder[i]);
}
root->left=buildTree(pre_left,vin_left);//递归将左右节点一一分开
root->right=buildTree(pre_right,vin_right);
return root;
}
};
思路:
由先序序列第一个pre[0]在中序序列中找到根节点位置gen
以gen为中心遍历
0~gen左子树
子中序序列:0~gen-1,放入vin_left[]
子先序序列:1~gen放入pre_left[],+1可以看图,因为头部有根节点
gen+1~vinlen为右子树
子中序序列:gen ~ vinlen-1放入vin_right[]
子先序序列:gen ~ vinlen-1放入pre_right[]
由先序序列pre[0]创建根节点
连接左子树,按照左子树子序列递归(pre_left[]和vin_left[])
连接右子树,按照右子树子序列递归(pre_right[]和vin_right[])
返回根节点
示意图如下:
另外补充:删除运算符不匹配错误
malloc对应free(c++中后面的主函数用的是delete,所以有此类错误)
new 对应delete
new是构造函数,与malloc只申请空间不同。
另外补充一点,因为每次都需要循环一遍中序数组,所以可以用打表法先把值储存起来,然后直接得到即可;