Skip to content

Commit 99df0d2

Browse files
committed
Add Trapping Rain Water solution in arrays
1 parent 788d95b commit 99df0d2

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
def trap(height):
2+
if not height:
3+
return 0
4+
left, right = 0, len(height) - 1
5+
max_left, max_right = 0, 0
6+
water = 0
7+
while left < right:
8+
if height[left] < height[right]:
9+
if height[left] >= max_left:
10+
max_left = height[left]
11+
else:
12+
water += max_left - height[left]
13+
left += 1
14+
else:
15+
if height[right] >= max_right:
16+
max_right = height[right]
17+
else:
18+
water += max_right - height[right]
19+
right -= 1
20+
return water

0 commit comments

Comments
 (0)