diff --git a/day01/day01.py b/day01/day01.py new file mode 100644 index 0000000..863fa7e --- /dev/null +++ b/day01/day01.py @@ -0,0 +1,45 @@ +import os + +def get_path(): + return os.path.join(os.path.dirname(__file__), 'data.txt') + +# Get the sum of the largest group +def solve_part1(): + [idx, value] = [0, 0] + [max_idx, max_value] = [0, 0] + + with open(get_path()) as fp: + for line in fp: + if line == '\n': + if value >= max_value: + [max_idx, max_value] = [idx, value] + [idx, value] = [idx+1, 0] + else: + value += int(line) + + print('largest sum #%s: %s' % (max_idx, max_value)) + +# Get the sum of the 3 largest groups +def solve_part2(): + largest = [0] + value = 0 + + with open(get_path()) as fp: + for line in fp: + if line == '\n': + largest.append(value) + largest.sort(reverse=True) + largest = largest[:3] + value = 0 + else: + value += int(line) + + print('largest 3 sums: %s, %s, %s' % (largest[0], largest[1], largest[2])) + print('total sum: %s' % sum(largest)) + +if __name__ == '__main__': + print('Part 1:') + solve_part1() + + print('Part 2:') + solve_part2() \ No newline at end of file