Update Insert into a Binary Search Tree.java

varunu28 · web-flow · commit 25de70cc455d · 2022-07-30T16:40:01.000-07:00
diff --git a/Medium/Insert into a Binary Search Tree.java b/Medium/Insert into a Binary Search Tree.java
@@ -12,16 +12,27 @@
  *         this.right = right;
  *     }
  * }
-*/
+ */
 class Solution {
   public TreeNode insertIntoBST(TreeNode root, int val) {
-    if (root == null) {
-      return new TreeNode(val);
-    } else if (root.val < val) {
-      root.right = insertIntoBST(root.right, val);
-    } else {
-      root.left = insertIntoBST(root.left, val);
+    TreeNode node = root;
+    while (node != null) {
+      if (val > node.val) {
+        if (node.right == null) {
+          node.right = new TreeNode(val);
+          return root;
+        } else {
+          node = node.right;
+        }
+      } else {
+        if (node.left == null) {
+          node.left = new TreeNode(val);
+          return root;
+        } else {
+          node = node.left;
+        }
+      }
     }
-    return root;
+    return new TreeNode(val);
   }
 }
