57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from functools import reduce
|
|
import numpy
|
|
|
|
rucksacks = []
|
|
|
|
input_file = open("input.txt", "r")
|
|
lines = input_file.readlines()
|
|
|
|
for line in lines:
|
|
line = line.replace("\n", "")
|
|
midpoint = int(len(line) / 2)
|
|
|
|
first_compartment = list(line[:midpoint])
|
|
second_compartment = list(line[midpoint:])
|
|
|
|
rucksacks.append([])
|
|
|
|
rucksacks[-1].append(first_compartment)
|
|
rucksacks[-1].append(second_compartment)
|
|
|
|
# part 1
|
|
|
|
|
|
def get_items_in_common(items_one, items_two):
|
|
items = [value for value in items_one if value in items_two]
|
|
return items
|
|
|
|
|
|
letters = list("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
|
|
|
|
total_priority_part_one = 0
|
|
for rucksack in rucksacks:
|
|
first_compartment = rucksack[0]
|
|
second_compartment = rucksack[1]
|
|
|
|
in_common = get_items_in_common(first_compartment, second_compartment)[0]
|
|
total_priority_part_one += letters.index(in_common) + 1
|
|
|
|
print("Part 1: " + str(total_priority_part_one))
|
|
|
|
# part 2
|
|
|
|
rucksack_groups = [[y[0]+y[1] for y in rucksacks][x:x+3]
|
|
for x in range(0, len(rucksacks), 3)]
|
|
|
|
total_priority_part_two = 0
|
|
for rucksack_group in rucksack_groups:
|
|
rucksack_one = rucksack_group[0]
|
|
rucksack_two = rucksack_group[1]
|
|
rucksack_three = rucksack_group[2]
|
|
|
|
in_common = reduce(numpy.intersect1d,
|
|
(rucksack_one, rucksack_two, rucksack_three))[0]
|
|
total_priority_part_two += letters.index(in_common) + 1
|
|
|
|
print("Part 2: " + str(total_priority_part_two))
|