third_max

a-abdellatif98 · Feb 16, 2022 · e97f39c · e97f39c
1 parent 6f5e5e9
commit e97f39c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/third_max.rb b/third_max.rb
@@ -1,3 +1,8 @@
 def third_max(nums)
-    return nums.uniq.sort[-3] 
+    sortnums = nums.uniq.sort
+    if sortnums.length < 3
+        return sortnums.last
+    else
+        return sortnums[-3]
+    end    
 end