CMU Coding Bootcamp
thecoded.prof
1from typing import List
2
3
4def largestColumnSum(L: List[List[int]]) -> int:
5 """Return the largest sum of any column in the matrix."""
6 cols = [sum(col) for col in zip(*L)]
7 return max(cols)
8
9
10def testLargestColumnSum():
11 print("Testing largestColumnSum()...", end="")
12 L = [[1, 2, 3], [4, 5, 6]]
13 assert largestColumnSum(L) == 9 # sum([3, 6])
14
15 L = [[3, 2, 1], [-5, -3, -6]]
16 assert largestColumnSum(L) == -1 # sum([2, -3])
17
18 L = [[1, 2, 3], [4, 5, 6], [7, -1, -2]]
19 assert largestColumnSum(L) == 12 # sum([1, 4, 7])
20
21 L = [[7, -1, -2]]
22 assert largestColumnSum(L) == 7 # sum([7])
23 print("Passed!")
24
25
26def main():
27 testLargestColumnSum()
28
29
30main()