抄算法笔记的代码的 很巧妙的思想
把镜像的有序二叉树看作preOrder中先访问right再访问left 妙!!
另外,这里用vector对比 先把所有结果列出来再来判断是查找二叉树还是镜像查找二叉树也是挺巧妙的
#include<cstdio>
#include<vector>
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);
}
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(){
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);
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;
}