889. 根据前序和后序遍历构造二叉树
前序和中序 道理相同 改变一些细节就可以了,在最后附上了前序中序的代码
题目描述
返回与给定的前序和后序遍历匹配的任何二叉树。
pre 和 post 遍历中的值是不同的正整数。
示例:
输入:pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
输出:[1,2,3,4,5,6,7]
提示:
1 <= pre.length == post.length <= 30
pre[] 和 post[] 都是 1, 2, …, pre.length 的排列
每个输入保证至少有一个答案。如果有多个答案,可以返回其中一个。
题解:
前序的第一个和后序的最后一个必定是根节点
前序的第二个节点可能是左子树的根节点也可能是右子树的根结点
如果是右子树的根结点,那就说明没有左子树,只有右子树,但是只有左子树和只有右子树的前序后序遍历出来是一样,所以默认是左子树的根节点
确定了根结点就要确定左子树区间长度,只要找到后序遍历中左子树根节点的位置idx,就可以确定下来长度为idx-postl
就是左子树区间长度,那么相应的右子树区间idx+1到postr
这样就可以根据前序和后序建立出二叉树
Ac代码
/**
* 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 {
public:
TreeNode* constructFromPrePost(vector<int>& pre, vector<int>& post) {
return helper(pre, 0, (int)pre.size() - 1, post, 0, (int