Update 0011-container-with-most-water.py

Clear optimized solution
Arrrsh · Apr 14, 2023 · 3bc43fe · 3bc43fe
1 parent 3433f21
commit 3bc43fe
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/0011-container-with-most-water.py b/python/0011-container-with-most-water.py
@@ -2,11 +2,15 @@ class Solution:
  def maxArea(self, height: List[int]) -> int:
  l, r = 0, len(height) - 1
  res = 0
+ h = max(height)
 
  while l < r:
  res = max(res, min(height[l], height[r]) * (r - l))
  if height[l] < height[r]:
  l += 1
  elif height[r] <= height[l]:
  r -= 1
+
+ if (r-l) * h <= res:
+ break 
  return res