给出中序遍历:[1,2,3]和前序遍历:[2,1,3]. 返回如下的树:
2
/ \
1 3
补充知识:
前序排列:根节点,左子节点,右子节点
中序排列:左子节点,根节点,右子节点
解题思路:
1.根据前序排列第一个数字确定根节点,并在中序排列中确定根节点位置
2.根据根节点的位置确定中序排列中左子树、右子树,中序排列中根节点之前的数字为左子树,根节点之后的部分为右子树
3.根据中序排列中左子树和右子树的长度确定在前序排列中的左子树、右子树
4.递归构建左子树、右子树,直到叶子节点(判断条件)
TreeNode * buildTree(vector<int> &preorder, vector<int> &inorder) {
// write your code here
TreeNode *root = NULL;
vector<int> preorder_l,preorder_r,inorder_l,inorder_r;
int i,root_index=0;
if(preorder.empty()!=1 || inorder.empty()!=1) {
root = new TreeNode(preorder[0]); // 在前序队列中找根节点
// 在中序队列中找出根节点位置
for(i=0; i<inorder.size(); i++) {
if(preorder[0] == inorder[i])
break;
root_index++;
}
// 左右子树的前序、中序队列
for(i=0; i<root_index; i++) {
preorder_l.push_back(preorder[i+1]);
inorder_l.push_back(inorder[i]);
}
for(i=root_index+1; i<inorder.size(); i++) {
preorder_r.push_back(preorder[i]);
inorder_r.push_back(inorder[i]);
}
//递归构建左右子树
root->left = buildTree(preorder_l, inorder_l);
root->right = buildTree(preorder_r, inorder_r);
}
return root;
}
类型 *p=new 类型[];//分配新的内存,指针指向数组第一个元素