import numpy as np
def parse_input(file_path):
with file_path.open("r") as fp:
banks = map(str.strip, fp.readlines())
return map(lambda x: list(map(int, list(x))), banks)
def max_jolt(bank, length):
if length==1:
return max(bank)
amax = np.argmax(bank[:-(length-1)])
return 10**(length-1)*bank[amax] + max_jolt(bank[amax+1:], length-1)
def solve_problem(file_name, length):
banks = parse_input(Path(cwd, file_name))
sumj = 0
for bank in banks:
sumj += max_jolt(bank, length)
return sumj
you are viewing a single comment's thread
view the rest of the comments
view the rest of the comments