对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数N(≤1000),第二行包含N个整数,为给出的整数键值序列,数字间以空格分隔。
输出格式:
输出的第一行首先给出判断结果,如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,则输出YES
,否侧输出NO
。如果判断结果是YES
,下一行输出对应二叉树的后序遍历序列。数字间以空格分隔,但行尾不能有多余的空格。
输入样例1:
7
8 6 5 7 10 8 11
结尾无空行
输出样例1:
YES
5 7 6 8 11 10 8
结尾无空行
输入样例2:
7
8 6 8 5 10 9 11
输出样例2:
NO
#include <bits/stdc++.h>
using namespace std;
int key[1000];
vector<int> tree[2];
bool judge(int root, int top, bool m){
// 左右子集下标
int left = root + 1, right = top;
if(m == true){
// 镜像 先序输出(根右左)
while(left <= top && key[left] >= key[root]) left++;
while(right > root && key[right] < key[root]) right--;
}
else{
// 非镜像(根左右)
while(left <= top && key[left] < key[root]) left++;
while(right > root && key[right] >= key[root]) right--;
}
// 类似二分查找 左右边界最后交叉
if(left - right != 1)
return false;
// 继续对左右子集判断 由于下标交叉 左子集上界为top=right 同理右子集根root=left
judge(root + 1, right, m);
judge(left, top, m);
// 后序输出
tree[m].push_back(key[root]);
return true;
}
int main() {
int n;
cin >> n;
for(int i = 0; i < n; i++)
cin >> key[i];
for(int mirror = 1; mirror >= 0; mirror--){
// BST判断 && 输出元素与输入元素数量相等
if(judge(0, n-1, mirror) == true && tree[mirror].size() == n){
cout << "YES" << endl << tree[mirror][0];
for(int i = 1; i < n; i++)
cout << " " << tree[mirror][i];
exit(0);
}
}
cout << "NO";
}
tree[i].size() == n
这行代码对应这组数据,不理解用来干什么的。
while (tmp[left] >= tmp[root] && left++ <= top) ;
while (tmp[right] < tmp[root] && right-- > root) ;
while (tmp[left] >= tmp[root] && left <= top) left++;
while (tmp[right] < tmp[root] && right > root) right--;
不晓得这两个有什么优先级的区别,上面那个会错,下面这个对了。应该都是判断后运算吧。