【问题描述】
给定二叉树节点个数以及其中序遍历和后序遍历,求其前序遍历。
【输入形式】
共三行。
第一行是一个正整数N,表示节点个数。
第二行共N个正整数,表示二叉树的中序遍历。
第三行共N个正整数,表示二叉树的后序遍历。
【输出形式】
共一行,输出前序遍历。
【样例输入】
10
3 6 7 4 5 9 10 2 8 1
3 7 6 5 10 9 4 8 1 2
【样例输出】
2 4 6 3 7 9 5 10 1 8
【数据范围】
N <= 10000
#include <iostream>
using namespace std;
struct node{
int data;
node* lchild;
node* rchild;
};
void PreOrder(node* root) {
if (root == NULL) return;
cout << root->data << ' ';
PreOrder(root->lchild);
PreOrder(root->rchild);
}
node* Create(int n, int midorder[], int postorder[])
{
node* root = NULL;
root = new node;
root->data = postorder[n - 1];
root->lchild = NULL; //非常重要!!!
root->rchild = NULL; //非常重要!!!
if (n == 1) return root;
int* mid1 = NULL, * mid2 = NULL, * post1 = NULL, * post2 = NULL, index = -1;
for (int i = 0; i < n; i++)
{
if (midorder[i] == root->data) { index = i; break; }
}
if (index > 0){
mid1 = new int[index], post1 = new int[index];
for (int i = 0; i < index; i++){
mid1[i] = midorder[i];
post1[i] = postorder[i];
}
}
if ((n - index - 1) > 0) {
mid2 = new int[n - index - 1], post2 = new int[n - index - 1];
for (int i = 0; i < n - index - 1; i++) {
mid2[i] = midorder[index + 1 + i];
post2[i] = postorder[index + i];
}
}
if ((n - index - 1) > 0) root->rchild = Create(n - index - 1, mid2, post2);
else root->rchild = NULL;
if (index > 0) root->lchild = Create(index, mid1, post1);
else root->lchild = NULL;
return root;
}
int main()
{
int n;
cin >> n;
int* midOrder = new int[n], * postOrder = new int[n];
for (int i = 0; i < n; i++)
{
cin >> midOrder[i];
}
for (int i = 0; i < n; i++)
{
cin >> postOrder[i];
}
node* root = Create(n, midOrder, postOrder);
PreOrder(root);
cout << endl;
delete[]midOrder; delete[]postOrder;
return 0;
}
反思:
1.指针需要初始化
(int* mid1 = NULL, * mid2 = NULL, * post1 = NULL, * post2 = NULL, index = -1;
root->lchild = NULL; root->rchild = NULL; )
我就是没有初始化root的左右子树……熬到了两点半!!!如果没有初始化,就存在野指针会导致preOrder出错。
2.数组会在不经意间越界!我的方法是拿【样例输入】来模拟一遍,仔细核对。
3.听从一位大神建议,学习单步调试。