牛客网刷题:【剑指offer】
1.题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
2.解题思路:从前序遍历可知,前序遍历数组pre的首元素就是二叉树的根结点,然后根据根结点的值在中序遍历中找到根结点的位置,那么根结点左边就为左子树的序列,根结点右边就是右子树的序列。
3.代码:
class Solution {
public:
TreeNode* rebuild(vector<int>& pre, int pre_left, int pre_right, vector<int>& vin, int vin_left, int vin_right)//建立二叉树,先定义前序遍历序列和中序遍历序列的左子树和右子树的数组//
{
if (pre_left > pre_right) return nullptr;//如果前序遍历序列的左子树大于右子树,就返回空//
TreeNode* root = new TreeNode(pre_left);//新的前序遍历的左子树的结点就是根结点//
for (int i=vin_left; i<=vin_right; ++i) //定义整型变量i为中序遍历的左子树,当i<vin_right 时++i, 循环 此算法//
{
if (vin[i] == root_val) //如果中序遍历i 等于根结点//
{
root->left = rebuild(pre, pre_left+1, pre_left+i-vin_left, vin, vin_left, i-1); //则 建立前序遍历的左子树加1,i-1 赋值给 根结点的左结点//
root->right = rebuild(pre, pre_left+i-vin_left+1, pre_right, vin, i+1, vin_right);// 把pre_left+i-vin_left+1, i+1, vin_right 赋值给根结点的右结点//
break; // 结束循环 //
}
}
return root; // 返回根结点 //
}
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)// 定义前序遍历和中序遍历的数组元素<int> //
{
return rebuild(pre, 0, pre.size()-1, vin, 0, vin.size()-1); // 返回建立的二叉树 //
}
}
4.总结:这一题是考察到数据结构中二叉树的应用,明显感受到自己对这一知识不掌握。刷完题后,觉得会了一点。要坚持刷题。加油!