1. 题目
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
2. 示例![](https://i-blog.csdnimg.cn/blog_migrate/4c02deacb33755baf6fd0ec9071d9341.png)
3. 分析
1.前序和中序可以构造二叉树,中序和后续也可以构造二叉树,但是前序和后序不能构造二叉树。
2.因为有中序就可以确定左子树和右子树区间。而用前序下标可以确定节点在根的左子树还是右子树。所以中序用区间,前序用下标
3.用前序第一个节点做根,再去中序里面找这个根节点,中序里这个节点的左区间就是它的左子树,右区间就是它的右子树,递归左子树与右子树,就可以构建这棵树。
递归展开图就不画了~自行完成理解
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* _buildTree(vector<int>& preorder, vector<int>& inorder, int& prei, int inbegin, int inend) {
if (inbegin > inend) return nullptr;
TreeNode* root = new TreeNode(preorder[prei]);
// 分割出左右子区间
int rooti = inbegin;
while (rooti <= inend){
if (inorder[rooti] == preorder[prei])
break;
else
rooti++;
}
++prei; // 遍历每颗子树的根
// [inbegin, rooti-1] rooti [rooti+1, inend]
root->left = _buildTree(preorder, inorder, prei, inbegin, rooti-1);
root->right = _buildTree(preorder, inorder, prei, rooti+1, inend);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int i = 0;
return _buildTree(preorder, inorder, i, 0, inorder.size()-1);
}
};