1043 Is It a Binary Search Tree (java)
判断给定的序列是否是二叉搜索树(对于任意节点左子树的所有节点值小于此节点值,任意右子树的节点值大于等于此节点值),或者是镜像的二叉搜索树(左右节点互换)
方法:对于给定的树的序列找到对应的右子树开始节点(搜索树就是第一个大于等于序列第一个数(因为数前序,第一个就是这个输的根节点),镜像则是小于他的数)再往后遍历找出是否有值是不符合规范的(搜索树:剩下的值都要大于等于根节点,镜像树:剩下的值都要小于根节点的值)。如果全部规范开始递归找他的左右子树是否规范。在递归的同时按照后序遍历的规则把节点值添加到list中。
代码:
package PAT2020;
import java.util.ArrayList;
import java.util.Scanner;
public class test {
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner s = new Scanner(System.in);
int n = s.nextInt();
int[] arr = new int[n];
if(n==0)
{
System.out.println("YES");
return;
}
for(int a=0;a<n;a++)
{
arr[a] = s.nextInt();
}
ArrayList<Integer> list = new ArrayList<Integer>();
if(isTree1(arr,0,n-1,list))
{
System.out.println("YES");
System.out.print(list.get(0));
for(int a=1;a<list.size();a++)
{
System.out.print(" "+list.get(a));
}
System.out.println();
}else if(isTree2(arr,0,n-1,list))
{
System.out.println("YES");
System.out.print(list.get(0));
for(int a=1;a<list.size();a++)
{
System.out.print(" "+list.get(a));
}
System.out.println();
}else
{
System.out.println("NO");
}
}
public static boolean isTree1(int[] arr,int left,int right,ArrayList<Integer> list)
{
if(left>right)
{
return true;
}
int a=left+1;
for(;a<=right && arr[left] > arr[a];a++);
int t = a;
for(;a<=right;a++)
{
if(arr[a]<arr[left])
{
return false;
}
}
boolean re = isTree1(arr,left+1,t-1,list) && isTree1(arr,t,right,list);
list.add(arr[left]);
if(!re)list.clear();
return re;
}
public static boolean isTree2(int[] arr,int left,int right,ArrayList<Integer> list)
{
if(left>right)
{
return true;
}
int a=left+1;
for(;a<=right && arr[left]<=arr[a];a++);
int t = a;
for(;a<=right;a++)
{
if(arr[left]<=arr[a])
{
return false;
}
}
boolean re = isTree2(arr,left+1,t-1,list) && isTree2(arr,t,right,list);
list.add(arr[left]);
if(!re)list.clear();
return re;
}
}