Advertisement
Jacob_Thomas

BST Insert

Jan 28th, 2021
674
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 1.48 KB | None | 0 0
  1. import java.util.*;
  2. class BstInsert{
  3.     public static void main(String args[]){
  4.         Node node = null;
  5.         Scanner sc = new Scanner(System.in);
  6.         System.out.println("Enter the size and keys to insert in BST");
  7.         int len = sc.nextInt();
  8.         int arr[] = new int[len];
  9.         for(int x = 0; x < len; x++){
  10.             arr[x] = sc.nextInt();
  11.             node = insert(node, arr[x]);
  12.         }
  13.  
  14.         while(true){
  15.             System.out.println("Press 1 for insertion");
  16.             System.out.println("Press 2 for traversal");
  17.             System.out.println("Press 3 to stop");
  18.             int x = sc.nextInt();
  19.             if(x == 1){
  20.                 System.out.println("Enter the key");
  21.                 insert(node, sc.nextInt());
  22.  
  23.             }
  24.             if(x == 2){
  25.                 traverse(node);
  26.             }
  27.             if(x == 3){
  28.                 break;
  29.             }
  30.         }
  31.  
  32.     }
  33.  
  34.     static Node insert(Node node, int key){
  35.         if(node == null){
  36.             node = new Node(key);
  37.             return node;
  38.         }
  39.         if(key < node.key){
  40.             node.left = insert(node.left, key);
  41.         }
  42.         else if(key > node.key){
  43.             node.right = insert(node.right, key);
  44.         }
  45.         return node;
  46.     }
  47.  
  48.     static void traverse(Node node){
  49.         if(node != null){
  50.             traverse(node.left);
  51.             System.out.println(node.key);
  52.             traverse(node.right);
  53.         }
  54.     }
  55.  
  56. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement