题目链接:https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/
题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
思路:一开始看到这个题,不知道是没信心的原因,还是以前做pat甲级题,遇到的都是根据前序中序转后序的题,很少有这种构建二叉树返回的题目,于是开始网上搜题解,但是搜了半天,题解都差强人意,写得很长并且难懂,于是凭借着乱搞的心态,居然自己写出来了,,,就是不断地去找根节点,然后构建出这个节点,再递归的去找这个节点的左子树和右子树。
代码:
/**
* 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 {
private:
vector<int> preorder, inorder;
public:
TreeNode* build(int root, int s, int e){
if(s > e) return NULL;
int i = s;
TreeNode* node = new TreeNode(preorder[root]);
while(inorder[i] != preorder[root]) i++;
node -> left = build(root + 1, s, i - 1);
node -> right = build(root + i - s + 1, i + 1, e);
return node;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
this -> preorder = preorder;
this -> inorder = inorder;
int n = preorder.size();
TreeNode *tree = build(0, 0, n - 1);
return tree;
}
};
Xinstein666 发布了12 篇原创文章 · 获赞 0 · 访问量 282 私信 关注