返回与给定的前序和后序遍历匹配的任何二叉树。
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 的排列
每个输入保证至少有一个答案。如果有多个答案,可以返回其中一个。
解题思路:
本题采用递归方式进行计算,由前序遍历和后序遍历特点可知:
前序遍历为:
(root node)->(preorder of left branch)->(preorder of right branch)
后序遍历为:
(post order of left branch)->(postorder of right branch)->(root node)
若最终二叉树被序列化表述为[1,2,3,4,5,6,7],那么其前序遍历为[1]+[2,4,5]+[3,6,7],而其后序遍历为[4,5,2]+[6,7,3]+[1].
如果知道左分支多少个点,就可以对这些数组进行分组,并用递归生成树的每个分支。
算法
令左分支有L个节点,左分支的头结点为pre[1],但它也出现在左分支的后序表示的最后,所以pre[1]=postL-1,因此当post[i]==pre[1]时,L=i+1;
由以上推论,在递归中左分支有pre[1:L+1]和post[0:L]重新分支,而右分支则由pre[L+1:N]和post[L:N-1]重新分支,即用一个左分支头结点把原数组分开表述。
c++代码:
class Solution
{
public:
TreeNode* constructFromPrePost(vector& pre, vector& post)
{
int N = pre.size();
if (N == 0)return NULL;
TreeNode* root = new TreeNode(pre[0]);
if (N == 1)return root;
int L = 0;
for (int i = 0; i < N; ++i)
if (post[i] == pre[1])
L = i + 1;
vector<int>pre_left(pre.begin() + 1, pre.begin() + L + 1);
vector<int>post_left(post.begin(), post.begin() + L);
vector<int>pre_right(pre.begin() + L + 1, pre.begin() + N);
vector<int>post_right(post.begin() + L, post.begin() + N - 1);
root->left = constructFromPrePost(pre_left, post_left);
root->right = constructFromPrePost(pre_right, post_right);
return root;
}
};