L3-010 是否完全二叉搜索树 (30分)java

将一系列给定数字顺序插入一个初始为空的二叉搜索树（定义为左子树键值大，右子树键值小），你需要判断最后的树是否一棵完全二叉树，并且给出其层序遍历的结果。

输入格式：
输入第一行给出一个不超过20的正整数N；第二行给出N个互不相同的正整数，其间以空格分隔。

输出格式：
将输入的N个正整数顺序插入一个初始为空的二叉搜索树。在第一行中输出结果树的层序遍历结果，数字间以1个空格分隔，行的首尾不得有多余空格。第二行输出YES，如果该树是完全二叉树；否则输出NO。

28分，两个测试点没过

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Vector;

public class Main {
    private static Vector<Integer> pre = new Vector<>();
    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        int n=Integer.parseInt(br.readLine());
        String s[]=br.readLine().split(" ");
        int[] arr=new int[n];
        int current=0,index=0;
        while(current<n){
            arr[current++]=Integer.parseInt(s[index++]);
        }
        TreeNode tr=new TreeNode();
        tr.data=arr[0];
        index=1;
        while(n-->1){
            tr.addNode1(arr[index++]);
        }
        cengxu(tr);
        check(tr);
    }
    private static void cengxu(TreeNode node) {
        if(node==null)return;
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(node);
        boolean flag=true;
        while(!queue.isEmpty()){
            TreeNode cur=queue.remove();
            if(flag){
                System.out.print(cur.data);
                flag=false;
            }
            else System.out.print(" "+cur.data);
            if(cur.left!=null){
                queue.add(cur.left);
            }
            if(cur.right!=null){
                queue.add(cur.right);
            }
        }
    }
    private static void check(TreeNode node) {
        boolean flag=true,flag1=true;
        if(node==null)return;
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(node);

        while(!queue.isEmpty()){
            TreeNode cur=queue.remove();


            if(cur.left!=null){
                queue.add(cur.left);
            }else {
                break;
            }

            if(cur.right!=null){
                queue.add(cur.right);
            }else {
                break;
            }
        }
        if (queue.peek().right == null){
            queue.poll();

        }
        //这里判断队列中剩余的节点是不是都是叶子节点
        while (!queue.isEmpty()){
            node = queue.poll();

            if (node.left != null || node.right != null){
                flag1 = false;
            }
        }
        System.out.println();
        if(flag1) System.out.println("YES");
        else System.out.println("NO");

    }
}
class TreeNode {
    int data;
    TreeNode left, right;

    public TreeNode() {
    }

    public TreeNode(int x) {
        data = x;
    }

    public void addNode(int num) {
        if (num < this.data) {
            if (this.left != null) {
                this.left.addNode(num);
            } else {
                this.left = new TreeNode(num);
            }
            return;
        }

        if (this.right != null) {
            this.right.addNode(num);
        } else {
            this.right = new TreeNode(num);
        }
        return;
    }

    public void addNode1(int num) {
        if (num >=this.data) {
            if (this.left != null) {
                this.left.addNode1(num);
            } else {
                this.left = new TreeNode(num);
            }
            return;
        }

        if (this.right != null) {
            this.right.addNode1(num);
        } else {
            this.right = new TreeNode(num);
        }
        return;
    }




}