123 lines
2.9 KiB
Python
123 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
|
|
from os.path import join, dirname
|
|
from typing import List
|
|
|
|
# Day 10
|
|
|
|
# Common
|
|
|
|
|
|
def read_input(filename):
|
|
data = join(dirname(__file__), '..', filename)
|
|
with open(data) as f:
|
|
return f.read().strip()
|
|
|
|
|
|
test = read_input('test.txt')
|
|
data = read_input('input.txt')
|
|
|
|
|
|
class Node:
|
|
def __init__(self, brackets, children):
|
|
self.brackets = brackets
|
|
self.children = children
|
|
|
|
def add_child(self, child):
|
|
self.children.append(child)
|
|
|
|
def __repr__(self) -> str:
|
|
return f'{self.brackets} -> {len(self.children)} {self.children}'
|
|
|
|
|
|
class Parser:
|
|
openers = ['(', '[', '{', '<']
|
|
closers = [')', ']', '}', '>']
|
|
|
|
def __init__(self, code: str):
|
|
self.tollerant = False
|
|
self.code = code
|
|
self.pos = 0
|
|
self.valid = True
|
|
self.nodes: List[Node] = []
|
|
|
|
def parse(self, tollerant=False):
|
|
self.tollerant = tollerant
|
|
root = Node(None, [])
|
|
try:
|
|
self.traverse(root)
|
|
except Exception as e:
|
|
# print(e.args[0])
|
|
pass
|
|
self.nodes.append(root)
|
|
|
|
def traverse(self, node):
|
|
c = self.code[self.pos]
|
|
typ = self.openers.index(c) if c in self.openers else -1
|
|
if typ == -1:
|
|
self.valid = False
|
|
raise Exception(
|
|
f'Invalid character at position {self.pos}. Found {c}')
|
|
|
|
node = Node(typ, [])
|
|
while True:
|
|
self.pos += 1
|
|
closing = self.closers[typ]
|
|
if self.pos >= len(self.code):
|
|
if self.tollerant:
|
|
self.code += closing
|
|
break
|
|
else:
|
|
raise Exception("EOF")
|
|
c = self.code[self.pos]
|
|
if c == closing:
|
|
break
|
|
else:
|
|
node.add_child(self.traverse(node))
|
|
return node
|
|
|
|
|
|
class Checker:
|
|
def __init__(self, lines: List[str]):
|
|
self.lines = [Parser(line) for line in lines]
|
|
|
|
def flag1(self):
|
|
for line in self.lines:
|
|
line.parse(tollerant=False)
|
|
points = {')': 3, ']': 57, '}': 1197, '>': 25137}
|
|
flag = [
|
|
points[line.code[line.pos]]
|
|
for line in self.lines
|
|
if not line.valid
|
|
]
|
|
return sum(flag)
|
|
|
|
def flag2(self):
|
|
added = ''
|
|
for line in self.lines:
|
|
print(f'\nChecking:')
|
|
print(line.code)
|
|
before = len(line.code)
|
|
line.parse(tollerant=True)
|
|
if line.valid:
|
|
print(line.code)
|
|
added += line.code[before:]
|
|
|
|
@ staticmethod
|
|
def parse(lines: str):
|
|
return Checker(lines.split('\n'))
|
|
|
|
|
|
# 1
|
|
print('1.')
|
|
checker = Checker.parse(test)
|
|
print(f'Test: {checker.flag1()}')
|
|
checker = Checker.parse(data)
|
|
print(f'Real: {checker.flag1()}')
|
|
|
|
# 2
|
|
print('\n2.')
|
|
|
|
checker = Checker.parse(test)
|
|
print(f'Test: {checker.flag2()}')
|