如果我要求解post-order(str1, str2)的话,首先不难发现,根据‘前序遍历’str1=‘根节点’+‘左子树的前序遍历’+‘右子树的前序遍历’,我可以知道这棵二叉树的根节点root便是str1的第一个字符!”小Ho道。
“而我在知道了‘根节点’root之后,我便可以利用‘中序遍历’str2=‘左子树的中序遍历’+‘根节点’+‘右子树的中序遍历’,求解出‘左子树的中序遍历’str2L和‘右子树的中序遍历’str2R!”
“接下来,由于一棵子树的前序遍历和中序遍历的长度相同,那么仍然是根据‘前序遍历’str1=‘根节点’+‘左子树的前序遍历’+‘右子树的前序遍历’,我可以知道从str1的第2个字符开始的str2L.length()个字符便是‘左子树的前序遍历’str1L,而这之后的部分便是‘右子树的前序遍历’str1R!”小Ho说到这里,顿了顿,希望从小Hi处得到些反馈。
“没错!那你准备如何求解post_order(str1, str2)呢?”小Hi肯定道。
“这只要根据之前求出的结果,和‘后序遍历’=‘左子树的后序遍历’+‘右子树的后序遍历’+‘根节点’,便可以知道——post_order(str1, str2)=post_order(str1l, str2l)+post_order(str1r, str2r)+root这样一个式子来!而这个问题的规模——也就是二叉树的大小,是一直在缩小的,所以是能够这样不断的划分做下去的!也就是说我之后可以将当前根节点的左子树又拆分成为两个问题来解决,一直到当前节点是叶子节点了为止!然后慢慢的将这些答案拼成上层的问题的答案,而这个过程只需要使用递归完成就可以了!”
#include<iostream>
#include<vector>
using namespace std;
vector<int>res;
void post_order(int *pre, int *in, int len)
{
if (len<1)return;
int i = 0; //每次递归i都初始为0
while (in[i] != pre[0])i++; //找出位置
post_order(pre + 1, in, i);
post_order(pre + i + 1, in + i + 1, len - i - 1);
res.push_back(pre[0]);
}
void pre_order(int *in, int *post, int len) {
if (len<1)return;
int i = 0; //每次递归i都初始为0
while (in[i] != post[len-1])i++; //找出位置
res.push_back(post[len - 1]);
pre_order(in, post, i);
pre_order(in+ i + 1, post + i, len - i - 1);
}
int main(){
int pre[] = { 8,7,5,4,6,3,2 };
int in[] = { 5,7,4,8,3,6,2 };
int post[] = { 5,4,7,3,2,6,8 };
post_order(pre, in, 7);
//pre_order(in, post, 7);
for (auto c : res)
cout << c << endl;
}