二叉排序树的实现

た入场券 2022-05-27 05:44 155阅读 0赞

> 题目描述

输入一系列整数，建立二叉排序数，并进行前序，中序，后序遍历。

> 输入描述:

输入第一行包括一个整数n(1<=n<=100)。  
接下来的一行包括n个整数。

> 输出描述:

可能有多组测试数据，对于每组数据，将题目所给数据建立一个二叉排序树，并对二叉排序树进行前序、中序和后序遍历。  
每种遍历结果输出一行。每行最后一个数据之后有一个空格。

输入中可能有重复元素，但是输出的二叉树遍历序列中重复元素不用输出。

> 示例1

输入

5  
1 6 5 9 8  
输出

1 6 5 9 8  
1 5 6 8 9  
5 8 9 6 1

import java.util.*;
    import java.io.*;
    class TreeNode{
        int val;
        TreeNode left;
        TreeNode right;
        public TreeNode(int val){
            this.val = val;
        }
    }
    public class Main{
        public static void main(String[] args){
            Scanner sc = new Scanner(System.in); 
            int n;
            while(sc.hasNextInt()){
                n = sc.nextInt();
                if(n < 0 || n == 0)
                    continue;
                int[] arr = new int[n];
                for (int i = 0; i < arr.length; i ++) {
                    arr[i] = sc.nextInt();
                }
                TreeNode pNode = new TreeNode(Integer.MAX_VALUE);
                TreeNode root = pNode;
                for(int i=0;i<n;i++){
                    pNode = root;
                    insertTreeNode(arr[i],pNode);
                }
                root = root.left;
                TreeNode temp = root;
                preOrder(temp);
                System.out.println();
                temp = root;
                InOrder(temp);
                System.out.println();
                temp = root;
                postOrder(temp);
                System.out.println();
            }
        }
        private static void preOrder(TreeNode pNode){
            if(pNode != null){
                System.out.print(pNode.val+" ");
                preOrder(pNode.left);
                preOrder(pNode.right);
            }
        }
        private static void InOrder(TreeNode pNode){
            if(pNode != null){
                InOrder(pNode.left);
                System.out.print(pNode.val+" ");
                InOrder(pNode.right);
            }
        }
        private static void postOrder(TreeNode pNode){
            if(pNode != null){
                postOrder(pNode.left);
                postOrder(pNode.right);
                System.out.print(pNode.val+" ");
            }
        }
        private static void insertTreeNode(int val,TreeNode pNode){
            if(pNode == null)
                pNode = new TreeNode(val);
            else if(val == pNode.val)
                return;
            else if(val > pNode.val){
                if(pNode.right != null){
                    insertTreeNode(val,pNode.right);
                }else{
                    TreeNode nNode = new TreeNode(val);
                    pNode.right = nNode;
                }
            }else{
                if(pNode.left != null){
                    insertTreeNode(val,pNode.left);
                }else{
                    TreeNode nNode = new TreeNode(val);
                    pNode.left = nNode;
                }
            }
        }
    }