Skip to content

Commit ad1ad7d

Browse files
Create RainWaterTapping.py
1 parent f886d7b commit ad1ad7d

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed
+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
# Rain water Trapping Problem
2+
3+
def rWTP(arr):
4+
left = []
5+
right = []
6+
n = len(arr)
7+
left.append(arr[0])
8+
right = [0 for i in range(n)]
9+
right[n-1] = arr[n-1]
10+
for i in range(1, n):
11+
left.append(max(left[i-1],arr[i]))
12+
for j in range(n-2, -1, -1):
13+
right.append(max(right[j+1], arr[j]))
14+
15+
ans = 0
16+
for i in range(n):
17+
ans += max(left[i], right[i]) - arr[i]
18+
19+
return ans
20+
21+
# Test 1:
22+
arr = [3, 1, 2, 4, 0, 1, 3, 2]
23+
print(rWTP(arr))

0 commit comments

Comments
 (0)