Files
aoc/03/c.py

34 lines
695 B
Python

import os
f = open(os.path.dirname(__file__)+"/input.txt", "r")
tot = 0
tot2 = 0
priorities = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
commons = ''
rucksacks = f.read().splitlines()
rucksacks2 = []
for data in rucksacks:
left = data[:len(data)//2]
right = data[len(data)//2:]
common = ''.join(set(left).intersection(right))
tot = tot + priorities.rfind(common)+1
print("First answer: " + str(tot))
start = 0
end = len(rucksacks)
step = 3
for i in range(start, end, step):
three = rucksacks[i:i+step]
common = set(three[0]) & set(three[1]) & set(three[2])
tot2 = tot2 + priorities.rfind(''.join(common))+1
print("Second answer: " + str(tot2))