PAT A1043 Is it a Binary Search Tree
思路:直接进行建树然后与输入序列进行对比,注意镜像序列即对换左右遍历顺序
#include<cstdio>
#include<vector>
using namespace std;
typedef struct Node Node;
struct Node {
int data;
Node* left, *right;
};
void insert(Node* &root, int data) {
if(root == NULL) {
root = new Node;
root->data = data;
root->left = NULL;
root->right = NULL;
return;
}
if(data < root->data) insert(root->left, data);
else insert(root->right, data);
}
void preOrder(Node* root, vector<int>& pre) {
if(root == NULL) return;
pre.push_back(root->data);
preOrder(root->left, pre);
preOrder(root->right, pre);
}
void mPreOrder(Node* root, vector<int>& mPre) {
if(root == NULL) return;
mPre.push_back(root->data);
mPreOrder(root->right, mPre);
mPreOrder(root->left, mPre);
}
void postOrder(Node* root, vector<int>& post) {
if(root == NULL) return;
postOrder(root->left, post);
postOrder(root->right, post);
post.push_back(root->data);
}
void mPostOrder(Node* root, vector<int>& mPost) {
if(root == NULL) return;
mPostOrder(root->right, mPost);
mPostOrder(root->left, mPost);
mPost.push_back(root->data);
}
int main() {
int n;
scanf("%d", &n);
int data;
Node* root = NULL;
vector<int> origin;
for(int i = 0; i < n; i++) {
scanf("%d", &data);
insert(root, data);
origin.push_back(data);
}
vector<int> pre, mPre, post, mPost;
preOrder(root, pre);
mPreOrder(root, mPre);
if(origin == pre) {
printf("YES\n");
postOrder(root, post);
for(int i = 0; i < post.size(); i++) {
printf("%d", post[i]);
if(i < post.size() - 1)
printf(" ");
else printf("\n");
}
} else if(origin == mPre) {
printf("YES\n");
mPostOrder(root, mPost);
for(int i = 0; i < mPost.size(); i++) {
printf("%d", mPost[i]);
if(i < mPost.size() - 1)
printf(" ");
else printf("\n");
}
} else {
printf("NO\n");
}
return 0;
}