renanvice-day-09-py

Month-1/Week-02/day-09/py/main.py

@@ -0,0 +1,24 @@
+def trappedWater(height):
+  area = 0
+  left = 0
+  right = left+1
+  maxr = len(height)-1
+  total = 0
+
+  while left < maxr:
+    if height[left] == 0:
+      left +=1
+      right = left+1
+    else:
+      area = right-left-1
+      if height[right] < height[left]:
+        if right == maxr:
+          height[left] = height[left] -1
+          right = left + 1
+          continue
+        right += 1
+      else:
+        total += area
+        height[left] = height[left] -1
+        right = left + 1
+  return total

Month-1/Week-02/day-09/py/main_test.py

@@ -0,0 +1,20 @@
+from main import trappedWater
+
+def trappedWater6():
+  assert trappedWater([0,1,0,2,1,0,1,3,2,1,2,1]) == 6
+
+def trappedWater9():
+  assert trappedWater([4,2,0,3,2,5]) == 9
+
+def trappedWater2():
+  assert trappedWater([1,0,1,0,1]) == 2
+
+def trappedWater7():
+  assert trappedWater([0,1,0,2,1,0,1,3,2,1,2,1,2]) == 7
+
+if __name__ == "__main__":
+  trappedWater6()
+  trappedWater9()
+  trappedWater2()
+  trappedWater7()
+  print('All test passed')