一棵二叉搜索树可被递归地定义为具有下列性质的二叉树:对于任一结点,
- 其左子树中所有结点的键值小于该结点的键值;
- 其右子树中所有结点的键值大于等于该结点的键值;
- 其左右子树都是二叉搜索树。
所谓二叉搜索树的“镜像”,即将所有结点的左右子树对换位置后所得到的树。
给定一个整数键值序列,现请你编写程序,判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。
输入格式:
输入的第一行给出正整数 N(≤1000)。随后一行给出 N 个整数键值,其间以空格分隔。
输出格式:
如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果,则首先在一行中输出 YES
,然后在下一行输出该树后序遍历的结果。数字间有 1 个空格,一行的首尾不得有多余空格。若答案是否,则输出 NO
。
输入样例 1:
7
8 6 5 7 10 8 11
输出样例 1:
YES
5 7 6 8 11 10 8
输入样例 2:
7
8 10 11 8 6 7 5
输出样例 2:
YES
11 8 10 7 5 6 8
输入样例 3:
7
8 6 8 5 10 9 11
输出样例 3:
NO
Note:
前序遍历:根左右,
后序遍历:左右根
二叉搜索树:最前面是根结点,中间一段比根节点小是左子树,最后一段比根节点大或等于是右子树
不需要建立树
1.用两各左右指针i,j分别指向左子树和右子树的根,之后分别遍历完左右子树,i和j恰好相差1,则是前序遍历,最后按照后序遍历进行递归(左右根)得到后序遍历结点值。
2.如果后序遍历结点个数不等于前序遍历结点个数,则可能是镜像二叉搜索树,如果是镜像二叉搜索树,则顺序就是根右左,与根节点比较大小恰好是与前序遍历相反的
3.遍历完之后,得到后序遍历的集合中的结点数如果还是不等于前序遍历结点数,则输出NO,否则就按顺序输出后序遍历的结点。注意空格
代码:
import java.util.Scanner;
import java.util.Vector;
public class Main {
private static boolean isMirror;//判断是否镜像
private static Vector<Integer> pre = new Vector<>();//存储前序结点
private static Vector<Integer> post = new Vector<>();//存储后序结点
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
for(int i = 0; i<n; i++)
pre.add(sc.nextInt());
getPost(0,n-1);
if(post.size() != n){//后序遍历数组长度与前序不一样,则是镜像二叉搜索树
isMirror = true;
post.clear();//清空vector集合
getPost(0,n-1);
}
if(post.size() == n){
System.out.println("YES");
for(int i = 0; i<n-1; i++){
System.out.print(post.get(i)+" ");
}
System.out.println(post.get(n-1));
}else
System.out.println("NO");
}
private static void getPost(int root, int tail) {
if(root > tail)return;//根节点小于右子树上的值
int i = root+1;
int j = tail;
if(!isMirror){
while(i <= tail && pre.get(root) > pre.get(i)){
i++;//二叉树的左子树均小于根节点
}
while (j>root && pre.get(root) <= pre.get(j)){
j--;//二叉树的右子树均大于等于根节点
}
}else{//若为镜像,根右左
while(i <= tail && pre.get(root) <= pre.get(i)){
i++;
}
while(j > root && pre.get(root) > pre.get(j)){
j--;
}
}
if(i-j != 1)return;
//遍历根节点左右子数,根后序遍历的顺序一样,左右根
getPost(root+1,j);//左
getPost(i,tail);//右
post.add(pre.get(root));//根
}
}