-
Notifications
You must be signed in to change notification settings - Fork 0
/
day_17.py
98 lines (78 loc) · 2.63 KB
/
day_17.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
from __future__ import division, print_function
import os
from my_utils.tests import test_function
def create_buff(step_size, buff_max=2017):
buff = [0]
curr_pos = 0
for buff_len in range(1, buff_max+1):
curr_pos = (curr_pos + step_size) % buff_len
curr_pos += 1
buff.insert(curr_pos, buff_len)
return buff
def simulate_buff(step_size, buff_max=2017):
zero_pos = 0
curr_pos = 0
val_after_zero = None
for buff_len in range(1, buff_max+1):
curr_pos = (curr_pos + step_size) % buff_len
curr_pos += 1
if curr_pos == zero_pos + 1:
val_after_zero = buff_len
if curr_pos <= zero_pos: # i.e. insert before if index same
zero_pos += 1
return val_after_zero
def part_1(step_size, buff_max=2017):
"""Function which calculates the solution to part 1
Arguments
---------
Returns
-------
"""
buff = create_buff(step_size, buff_max=buff_max)
idx = buff.index(buff_max)
return buff[idx + 1 % len(buff)]
def part_2(step_size):
"""Function which calculates the solution to part 2
Arguments
---------
Returns
-------
"""
return simulate_buff(step_size, int(50*1e6))
def main(test_datas, functions, puzzle_input=None, test_functions=None):
if test_functions is None:
test_functions = functions
for ii, (test_data, fun) in enumerate(zip(test_datas, test_functions)):
nr_errors = test_function(fun, test_data)
if nr_errors == 0:
print('Pt. {} Tests Passed'.format(ii+1))
if puzzle_input is not None:
fn = os.path.basename(__file__)
for ii, fun in enumerate(functions):
ans = fun(puzzle_input)
print('{} Pt. {} Solution: {}'.format(fn, ii+1, ans))
if __name__ == "__main__":
# Testing data:
# - each element of input list will be passed to function
# - the relative element in output list is the expected output
test_data1 = {
'inputs': [3],
'outputs': [638]
}
test_data2 = {
'inputs': [],
'outputs': []
}
# Code to import the actual puzzle input
with open('./inputs/day_17.txt') as f:
puzzle_input = f.read().strip()
puzzle_input = int(puzzle_input)
# puzzle_input = [line.rstrip('\n') for line in f]
# Main call: performs testing and calculates puzzle outputs
main(test_datas=[],
functions=[part_2],
puzzle_input=puzzle_input,
test_functions=None)
# main(test_datas=[test_data1, test_data2],
# functions=[part_1, part_2],
# puzzle_input=puzzle_input)