题目大意是:定义BST,其左子树都是小于父亲节点的值,右子树都是不小于父亲节点的值。以及镜像BST, 其左子树都是不小于父亲节点的值,右子树都是小于父亲节点的值。然后给出一个序列,判断是不是BST或者镜像BST的先序遍历序列。
对于这个问题,首先我看到BST就想到中序序列有序了,然后想能不能通过先序和中序序列来构造一棵树,再判断是不是BST。但很不幸,犯了一个错误,如果不是BST的话,中序序列就不是我排序后的序列了,这样根本不能构造一棵树。
看了别人的思路,解决这个问题关键要想到按照给出的序列来直接构造BST和镜像BST。以构造BST为例,如果构造出的真的是BST,那么再进行一次先序遍历,得到的序列应该是和给出序列一致的。如果不一致,那它就不是BST。对于镜像BST亦然。只是构造镜像BST时要注意构造方法是小于父亲节点的值放在右子树,不小于父亲节点的值放在左子树。另外也可以只构造BST,只需要改变先序遍历的顺序(即先访问右子树,再访问左子树,最后访问根节点)就可以得到对应的镜像BST的先序遍历序列。
总之,感觉这个思路对我来说还是一时想不到的,相当于 通过 BST的构造方法 和 假定的先序序列, 来构造BST, 再反过来检验 BST的先序序列是否一致。
AC代码如下:
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cstring>
using namespace std;
struct Node
{
int data;
Node* left;
Node* right;
Node(int data)
{
this->data = data;
this->left = this->right = NULL;
}
};
void insert(Node* &node, int data)
{
if(node == NULL)
{
node = new Node(data);
return;
}
if(data < node->data) insert(node->left, data);
else insert(node->right, data);
}
void mirrorInsert(Node* &node, int data)
{
if(node == NULL)
{
node = new Node(data);
return;
}
if(data >= node->data) mirrorInsert(node->left, data);
else mirrorInsert(node->right, data);
}
void preOrder(vector<int> &pre, Node* node)
{
if(node)
{
pre.push_back(node->data);
preOrder(pre, node->left);
preOrder(pre, node->right);
}
}
void postOrder(Node* node, int count)
{
static int printCount = 0;
if(node)
{
postOrder(node->left, count);
postOrder(node->right, count);
printf("%d", node->data);
printCount++;
if(printCount < count) printf(" ");
}
}
int main()
{
Node *root = NULL, *mirrorRoot = NULL;
int N;
cin >> N;
vector<int> v(N);
for (int i = 0; i < N; ++i)
{
scanf("%d", &v[i]);
insert(root, v[i]);
mirrorInsert(mirrorRoot, v[i]);
}
vector<int> pre, mirrorPre;
preOrder(pre, root);
if(pre == v)
{
printf("YES\n");
postOrder(root, N);
}
else
{
preOrder(mirrorPre, mirrorRoot);
if(mirrorPre == v)
{
printf("YES\n");
postOrder(mirrorRoot, N);
}
else printf("NO\n");
}
return 0;
}