-
Notifications
You must be signed in to change notification settings - Fork 0
/
aoc01.py
103 lines (82 loc) · 1.95 KB
/
aoc01.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
99
100
101
102
103
########################################################################
# Advent of Code 2022 - solver
#
# Copyright (C) 2022 Antonio Ceballos Roa
########################################################################
def parse_input(input_str):
parsed_input = input_str.strip().split('\n')
parsed_input.append('')
return parsed_input
########################################################################
# Algorithms
########################################################################
def solve_1(input_str):
parsed_input = parse_input(input_str)
calories = 0
max_calories = 0
for line in parsed_input:
try:
c = int(line)
calories += c
except ValueError:
if calories > max_calories:
max_calories = calories
calories = 0
return max_calories
def solve_2(input_str):
parsed_input = parse_input(input_str)
calories = 0
elf_calories = []
for line in parsed_input:
try:
c = int(line)
calories += c
except ValueError:
elf_calories.append(calories)
calories = 0
max_calories = sum(list(reversed(sorted(elf_calories)))[0:3])
return max_calories
########################################################################
# Test class
########################################################################
import unittest
class TestAoc(unittest.TestCase):
def setUp(self):
self.tc_1 = [
(
"""
1000
2000
3000
4000
5000
6000
7000
8000
9000
10000
""", 24000),
]
self.tc_2 = [
(
"""
1000
2000
3000
4000
5000
6000
7000
8000
9000
10000
""", 45000),
]
def tearDown(self):
pass
def test_solve_1(self):
for t in self.tc_1:
self.assertEqual(solve_1(t[0]), t[1])
def test_solve_2(self):
for t in self.tc_2:
self.assertEqual(solve_2(t[0]), t[1])