from math import ceil def numberOfBricks(steps: int) -> int: """Calculate the number of bricks in some steps using arithmetic series: S(n) = n(n+1)/2.""" return ceil((steps * (steps + 1)) / 2) print("Testing numberOfBricks()...", end="") assert numberOfBricks(0) == 0 assert numberOfBricks(1) == 1 assert numberOfBricks(2) == 3 assert numberOfBricks(3) == 6 assert numberOfBricks(4) == 10 assert numberOfBricks(10) == 55 print("Passed!")