diff --git a/src/com/dsa/trees/LevelOrderTraversal.java b/src/com/dsa/trees/LevelOrderTraversal.java index 896e1d1..a200ee0 100644 --- a/src/com/dsa/trees/LevelOrderTraversal.java +++ b/src/com/dsa/trees/LevelOrderTraversal.java @@ -32,6 +32,7 @@ public class LevelOrderTraversal { for (List level : list) { System.out.println(level); } + System.out.println("MinDepth: " + minDepth(root)); } catch (IOException e) { throw new RuntimeException(e); } @@ -74,4 +75,13 @@ public class LevelOrderTraversal { } return res; } + + public static int minDepth(TreeNode root) { + if(root == null) return 0; + int leftHeight = minDepth(root.left); + int rightHeight = minDepth(root.right); + if (leftHeight == 0) return rightHeight + 1; + if (rightHeight == 0) return leftHeight + 1; + return Math.min(leftHeight, rightHeight) + 1; + } }