543

javiermon · Jul 8, 2022 · ca97a48 · ca97a48
1 parent 742ae2e
commit ca97a48
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/ruby/543-Diameter-of-Binary-Tree.rb b/ruby/543-Diameter-of-Binary-Tree.rb
@@ -0,0 +1,16 @@
+def diameter_of_binary_tree(root)
+ $max = 0
+ max_height = diameter(root)
+ $max > max_height ? $max : max_height
+end
+
+def diameter(root)
+ return -1 if root.nil?
+
+ left = 1 + diameter(root.left)
+ right = 1 + diameter(root.right)
+ diameter = left + right
+ $max = diameter if diameter > $max
+
+ left > right ? left : right
+end