103 lines
2.3 KiB
Python
103 lines
2.3 KiB
Python
#!/bin/python3
|
|
import sys,time,re,os,copy
|
|
from pprint import pprint
|
|
sys.path.insert(0, '../../')
|
|
from fred import *
|
|
|
|
import math
|
|
|
|
start_time = time.time()
|
|
|
|
if sys.argv[1] == 'test':
|
|
input_f = 'test'
|
|
elif sys.argv[1] == 'input':
|
|
input_f = 'input'
|
|
else:
|
|
print('No argv provided')
|
|
exit()
|
|
|
|
|
|
#########################################
|
|
# #
|
|
# Part 1 #
|
|
# #
|
|
#########################################
|
|
|
|
def part1():
|
|
|
|
score = 0
|
|
grid = []
|
|
lines = loadFile(input_f)
|
|
|
|
pattern = re.compile(r'\d*[^\s]|\*|\+')
|
|
|
|
for l in lines:
|
|
p = pattern.findall(l)
|
|
try: #i want my numbers to be integers unless its the last row of operations.
|
|
grid.append(list2int(p))
|
|
except:
|
|
grid.append(p)
|
|
|
|
numbers = list(map(list, zip(*grid[:-1])))
|
|
ops = grid[-1]
|
|
|
|
|
|
for x in range(len(ops)):
|
|
if ops[x] == '*':
|
|
|
|
score += math.prod(numbers[x])
|
|
else:
|
|
score += sum(numbers[x])
|
|
|
|
return score
|
|
|
|
start_time = time.time()
|
|
p1 = part1()
|
|
print('Part 1:',p1, '\t\t', round((time.time() - start_time)*1000), 'ms')
|
|
|
|
#########################################
|
|
# #
|
|
# Part 2 #
|
|
# #
|
|
#########################################
|
|
|
|
def part2():
|
|
score = 0
|
|
|
|
grid = []
|
|
lines = []
|
|
with open(input_f) as file:
|
|
for line in file:
|
|
lines.append(line.rstrip('\n')) #my loadFile helper stripped trailing spaces and newline. We just want to remove newline.
|
|
|
|
for l in lines:
|
|
p = list(l)
|
|
grid.append(p)
|
|
|
|
numbers = list(map(list, zip(*grid[:-1])))
|
|
ops = [x.strip(' ') for x in grid[-1] if x != ' ']
|
|
count = 0
|
|
tmp = 0
|
|
for x in range(len(numbers)):
|
|
if all( n == ' ' for n in numbers[x]):
|
|
score += tmp
|
|
tmp = 0
|
|
count += 1
|
|
continue
|
|
op = ops[count]
|
|
|
|
if op == '+':
|
|
tmp += int(''.join(numbers[x]))
|
|
else:
|
|
if tmp == 0:
|
|
tmp = 1
|
|
tmp *= int(''.join(numbers[x]))
|
|
score += tmp
|
|
|
|
return score
|
|
|
|
start_time = time.time()
|
|
p2 = part2()
|
|
print('Part 2:',p2, '', round((time.time() - start_time)*1000), 'ms')
|
|
|