Resolved 'Trapping Rain Water' LeetCode problem with an optimized solution

Author: DevOgabek

Python/trapping_rain_water.py

@@ -0,0 +1,25 @@
+# Trapping Rain Water
+
+class Solution(object):
+    def trap(self, height):
+        left, right = 0, len(height) - 1
+        left_max = right_max = water_trapped = 0
+        
+        while left < right:
+            if height[left] <= height[right]:
+                if height[left] > left_max:
+                    left_max = height[left]
+                else:
+                    water_trapped += left_max - height[left]
+                left += 1
+            else:
+                if height[right] > right_max:
+                    right_max = height[right]
+                else:
+                    water_trapped += right_max - height[right]
+                right -= 1
+        
+        return water_trapped
+    
+solution = Solution()
+print(solution.trap([0,1,0,2,1,0,1,3,2,1,2,1]))