99 lines
2.6 KiB
Python
99 lines
2.6 KiB
Python
#!/bin/python3
|
|
import sys,time,re
|
|
from sympy import *
|
|
from pprint import pprint
|
|
sys.path.insert(0, '../../')
|
|
from fred import list2int,get_re,nprint,lprint,loadFile
|
|
start_time = time.time()
|
|
|
|
input_f = 'input'
|
|
|
|
#########################################
|
|
# #
|
|
# Part 1 #
|
|
# #
|
|
#########################################
|
|
def part1():
|
|
instructions = []
|
|
|
|
with open(input_f) as file:
|
|
for line in file:
|
|
match = get_re(r".*X[\+|=](\d+), Y[\+|=](\d+)",line.rstrip())
|
|
if match:
|
|
instructions.append(list2int([match.group(1),match.group(2)]))
|
|
|
|
a_cost = 3
|
|
b_cost = 1
|
|
|
|
a = ()
|
|
b = ()
|
|
r = ()
|
|
x,y = symbols('x y',integer=True)
|
|
result = 0
|
|
|
|
for idx, inst in enumerate(instructions):
|
|
|
|
if idx%3 == 0:
|
|
a = (inst[0],inst[1])
|
|
elif idx%3 == 1:
|
|
b = (inst[0],inst[1])
|
|
elif idx%3 == 2:
|
|
r = (inst[0],inst[1])
|
|
eq1 = Eq(a[0]*x + b[0]*y,r[0])
|
|
eq2 = Eq(a[1]*x + b[1]*y,r[1])
|
|
|
|
s = solve((eq1,eq2),(x,y))
|
|
if s:
|
|
result += s[x]*a_cost
|
|
result += s[y]*b_cost
|
|
return result
|
|
|
|
start_time = time.time()
|
|
#rint('Part 1:',part1(), '\t\t', round((time.time() - start_time)*1000), 'ms')
|
|
|
|
|
|
#########################################
|
|
# #
|
|
# Part 2 #
|
|
# #
|
|
#########################################
|
|
def part2():
|
|
instructions = []
|
|
|
|
with open(input_f) as file:
|
|
for line in file:
|
|
match = get_re(r".*X[\+|=](\d+), Y[\+|=](\d+)",line.rstrip())
|
|
if match:
|
|
instructions.append(list2int([match.group(1),match.group(2)]))
|
|
|
|
a_cost = 3
|
|
b_cost = 1
|
|
|
|
a = ()
|
|
b = ()
|
|
r = ()
|
|
x,y = symbols('x y',integer=True)
|
|
result = 0
|
|
|
|
for idx, inst in enumerate(instructions):
|
|
|
|
if idx%3 == 0:
|
|
a = (inst[0],inst[1])
|
|
elif idx%3 == 1:
|
|
b = (inst[0],inst[1])
|
|
elif idx%3 == 2:
|
|
r = (inst[0]+10000000000000,inst[1]+10000000000000)
|
|
eq1 = Eq(a[0]*x + b[0]*y,r[0])
|
|
eq2 = Eq(a[1]*x + b[1]*y,r[1])
|
|
|
|
s = solve((eq1,eq2),(x,y))
|
|
if s:
|
|
result += s[x]*a_cost
|
|
result += s[y]*b_cost
|
|
#input()
|
|
return result
|
|
|
|
start_time = time.time()
|
|
|
|
start_time = time.time()
|
|
print('Part 2:',part2(), '\t\t', round((time.time() - start_time)*1000), 'ms') |