61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os.path
|
|
|
|
import pytest
|
|
|
|
import support
|
|
|
|
INPUT_TXT = os.path.join(os.path.dirname(__file__), 'input.txt')
|
|
|
|
from string import ascii_letters
|
|
|
|
|
|
def compute(s: str) -> int:
|
|
total_priorities = 0
|
|
for line in s.split('\n'):
|
|
print(line)
|
|
c1, c2 = line[:int(len(line) / 2)], line[int(len(line)/2):]
|
|
common = None
|
|
for char in c1:
|
|
if char in c2:
|
|
common = char
|
|
break
|
|
if common is None:
|
|
continue
|
|
total_priorities += ascii_letters.find(common) + 1
|
|
|
|
return total_priorities
|
|
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('input_s', 'expected'),
|
|
(
|
|
('vJrwpWtwJgWrhcsFMMfFFhFp', ascii_letters.find('p') + 1),
|
|
('jqHRNqRjqzjGDLGLrsFMfFZSrLrFZsSL', ascii_letters.find('L') + 1),
|
|
('PmmdzqPrVvPwwTWBwg', ascii_letters.find('P') + 1),
|
|
('wMqvLMZHhHMvwLHjbvcjnnSBnvTQFn', ascii_letters.find('v') + 1),
|
|
('ttgJtRGJQctTZtZT', ascii_letters.find('t') + 1),
|
|
('CrZsJsPPZsGzwwsLwLmpwMDw', ascii_letters.find('s') + 1),
|
|
)
|
|
)
|
|
def test(input_s: str, expected: int) -> None:
|
|
assert compute(input_s) == expected
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('data_file', nargs='?', default=INPUT_TXT)
|
|
args = parser.parse_args()
|
|
|
|
with open(args.data_file) as f, support.timing():
|
|
print(compute(f.read()))
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main())
|