1. 题目来自于PAT A 1043,问题的大概描述如下:
给出一个N个正整数来作为一棵二叉排序树的节点插入顺序,问:这串序列是否是该二叉树的先序序列或者是二叉排序树的镜像树的先序序列,镜像树指的是交换根节点的左右子树形成的二叉树,如果对于输入的序列与二叉排序树的先序序列是一样的那么输出YES,并且输出二叉排序树的后序遍历的序列,如果输入的序列与二叉排序树的镜像树的先序序列是一致的,那么输出YES,并且输出镜像树的后序遍历的序列,其余情况输出NO
2. 思路分析:
① 首先给定插入的序列,我们要创建出这颗二叉排序树,二叉排序树的根节点的左子树的所有节点的数据域均小于等于根节点的数据域,右子树所有节点的数据域大于根节点的数据域
② 使用vector来存放初始序列、先序序列、镜像先序序列,可以很方便地相互进行比较,如果使用数组,则比较操作就需要使用循环来进行实现
③ 需要注意的问题:我们在定义节点的时候要设其为空节点(一开始是没有元素的)在新建节点的时候要注意令其左右节点地址设置为NULL
使用了vector之后那么可以很方便地进行比较然后输出就可以了
3. 下面是具体的C++代码:
#include<cstdio>
#include<vector>
#include<iostream>
using namespace std;
struct node{
int data;
node *left;
node *right;
};
void insert(node *&root, int data){
if(root == NULL){
//到达空节点的时候说明即为需要插入的位置
root = new node;
root->data = data;
root->left = root->right = NULL;
return;
}
if(data < root->data) insert(root->left, data);
else{
insert(root->right, data);
}
}
void preOrder(node *root, vector<int> &vi){
if(root == NULL) return;
vi.push_back(root->data);;
preOrder(root->left, vi);
preOrder(root->right, vi);
}
//镜像树先序遍历,记过存放在vi中
void preOrderMirror(node *root, vector<int> &vi){
if(root == NULL) return;
vi.push_back(root->data);
preOrderMirror(root->right, vi);
preOrderMirror(root->left, vi);
}
void postOrder(node *root, vector<int> &vi){
if(root == NULL) return;
postOrder(root->left, vi);
postOrder(root->right, vi);
vi.push_back(root->data);
}
void postOrderMirror(node *root, vector<int> &vi){
if(root == NULL) return;
postOrderMirror(root->right, vi);
postOrderMirror(root->left, vi);
vi.push_back(root->data);
}
vector<int> origin, pre, preM, post, postM;
int main(void){
int n, data;
node *root = NULL;
scanf("%d", &n);
for(int i = 0; i < n; ++i){
scanf("%d", &data);
origin.push_back(data);
insert(root, data);
}
preOrder(root, pre);
preOrderMirror(root, preM);
postOrder(root, post);
postOrderMirror(root, postM);
//对于vector数组可以直接使用==来进行比较
if(origin == pre){
printf("YES\n");
for(int i = 0; i < post.size(); ++i){
printf("%d", post[i]);
if(i < post.size() - 1) printf(" ");
}
}else if(origin == preM){
printf("YES\n");
for(int i = 0; i < postM.size(); ++i){
printf("%d", postM[i]);
if(i < postM.size() - 1) printf(" ");
}
}else{
printf("NO\n");
}
return 0;
}