总结的部分题目思路与代码,待完善。
【剑指offer-第二版】部分题目与解答【C++版本】
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树(假设没有重复数字)。树节点定义如下:
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) :val(x), left(nullptr), right(nullptr) {}
};
1.前序遍历的第一个数字总是树的根节点的值。但在中序遍历中,根节点的值在序列的中间,其中左子树的节点的值位于根节点的值的左边,右子树的节点的值位于根节点的值的右边。要扫描中序遍历序列,才能找到根节点的值。
2.又因为前序遍历总是在根节点后先遍历完左子树,才会遍历右子树,所以由后续遍历推断出左子树中节点的数目之后,前序遍历中根节点之后的相同数目的值都是左子树的节点的值。剩下的就是右子树的值。这样就分别找到了左右子树序列。
3.分别确定了根节点和左右子树的前序、中序遍历,我们可以用同样的方法构建左右子树,剩下的可以用递归来完成。
#include
#include
#include
#include
#include
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) :val(x), left(nullptr), right(nullptr) {}
};
TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin);
int main()
{
vector<int> pre{1,2,4,5,3,6,7};
vector<int> vin{4,2,5,1,6,3,7};
TreeNode *newRoot = reConstructBinaryTree(pre, vin);
system("pause");
return 0;
}
TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin)
{
if (pre.size() == 0 || pre.size() != vin.size())
return nullptr;
int root = pre[0];
TreeNode *newNode = new TreeNode(root);
//如果只剩一个节点了,那么可以直接返回
if (pre.size() == 1)
return newNode;
auto posi = find(vin.begin(), vin.end(), root);
//错误检测
if (posi == vin.end()) {
return nullptr;
}
int leftSize = posi - vin.begin();
int rightSize = vin.end() - posi - 1;
//递归求解
//这里取前序和后序遍历的左右子树可能有点绕,可以好好思考一下
newNode->left = reConstructBinaryTree(vector<int>(pre.begin() + 1, pre.begin() + 1 + leftSize),
vector<int>(vin.begin(), vin.begin() + leftSize));
newNode->right = reConstructBinaryTree(vector<int>(pre.begin() + 1 + leftSize, pre.end()),
vector<int>(vin.begin() + leftSize + 1, vin.end()));
return newNode;
}