CMU Coding Bootcamp
thecoded.prof
1from math import ceil
2
3
4def numberOfBricks(steps: int) -> int:
5 """Calculate the number of bricks in some steps using arithmetic series: S(n) = n(n+1)/2."""
6 return ceil((steps * (steps + 1)) / 2)
7
8
9print("Testing numberOfBricks()...", end="")
10assert numberOfBricks(0) == 0
11assert numberOfBricks(1) == 1
12assert numberOfBricks(2) == 3
13assert numberOfBricks(3) == 6
14assert numberOfBricks(4) == 10
15assert numberOfBricks(10) == 55
16print("Passed!")