leetcode106. 从中序与后序遍历序列构造二叉树
给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。
示例 1:
输入:inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出:[3,9,20,null,null,15,7]
示例 2:
输入:inorder = [-1], postorder = [-1]
输出:[-1]
算法思路
中序遍历:左子树、根节点、右子树。
后序遍历:左子树、右子树、根节点。
利用这两个特点,可以:
从后序数组中找到最后一个元素,这将是树的根节点。
在中序数组中找到根节点的位置,从而确定左子树和右子树的边界。
递归地对左子树和右子树进行同样的操作。
具体代码逻辑分析
首先检查后序数组是否为空,如果为空则返回NULL。
获取后序数组的最后一个元素作为根节点的值,并创建根节点。
处理特殊情况:
如果后序数组只有一个元素,直接返回根节点。
寻找根节点在中序数组中的位置:
通过遍历中序数组找到根节点的位置,变量del记录这个位置。
分割子数组:
根据del的值,将中序数组和后序数组分割为左子树和右子树的中序和后序数组。
递归构建左右子树:
递归调用traversal函数,分别构建左子树和右子树。
具体代码
class Solution {
public:TreeNode* traversal(vector<int>& inorder,vector<int>& postorder){if(postorder.size()==0) return NULL;int rootValue=postorder[postorder.size()-1];TreeNode* root=new TreeNode(rootValue);if(postorder.size()==1) return root;int del;for(del=0;del<inorder.size();del++){if(inorder[del]==rootValue) break;}vector<int> leftInorder(inorder.begin(),inorder.begin()+del);vector<int> rightInorder(inorder.begin()+del+1,inorder.end());vector<int> leftPostorder(postorder.begin(),postorder.begin()+leftInorder.size());vector<int> rightPostorder(postorder.begin()+leftInorder.size(),postorder.end()-1);root->left=traversal(leftInorder,leftPostorder);root->right=traversal(rightInorder,rightPostorder);return root;}TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {if(inorder.size()==0 || postorder.size()==0) return NULL;return traversal(inorder,postorder);}
};