#!/bin/python3 import sys from pprint import pprint import numpy as np input_f = sys.argv[1] arr = [] from aocd import get_data #print(get_data(day=6,year=2023)) maps = open(input_f).read().splitlines() maps = [l.split() for l in maps] pprint(maps) total_winnings = [] for i in range(0,len(maps[0])): if i == 0: continue time = int(maps[0][i]) max_dist = int(maps[1][i]) hold = 0 dist = 0 total_winnings.append(0) print(total_winnings) print('Start\t',end='') print(maps[0][i],maps[1][i]) print('Time\tDist\tHold') for j in range(1,int(maps[0][i])+1): while time > 0: time -= 1 hold += 1 dist = hold * time won = False print(str(time)+ '\t' + str(dist) + '\t' + str(hold) + '\t',end='') if dist > max_dist: total_winnings[i-1] += 1 print('won') else: print() print(np.prod(total_winnings)) print(total_winnings)