Move python solutions to separate directory
This commit is contained in:
parent
47e97f4533
commit
3903907973
70 changed files with 0 additions and 0 deletions
|
|
@ -1,95 +0,0 @@
|
|||
import sys
|
||||
|
||||
# PART 1
|
||||
|
||||
class Stream:
|
||||
def __init__(self, string):
|
||||
self.content = list(string)
|
||||
self.content.reverse()
|
||||
|
||||
def peek(self):
|
||||
return self.content[-1]
|
||||
|
||||
def pop(self):
|
||||
return self.content.pop()
|
||||
|
||||
def load_groups(filename):
|
||||
groups = []
|
||||
with open(filename, "r") as f:
|
||||
for line in f:
|
||||
stream = Stream(line[:-1])
|
||||
group, garbage = parse_group(stream)
|
||||
groups.append((group, garbage))
|
||||
return groups
|
||||
|
||||
def parse_group(stream):
|
||||
assert stream.pop() == "{"
|
||||
|
||||
if stream.peek() == "}":
|
||||
assert stream.pop() == "}"
|
||||
return [], 0
|
||||
|
||||
groups = []
|
||||
garbage = 0
|
||||
|
||||
while True:
|
||||
# Determine which sub-parser to use
|
||||
if stream.peek() == "{":
|
||||
group, more_garbage = parse_group(stream)
|
||||
groups.append(group)
|
||||
garbage += more_garbage
|
||||
elif stream.peek() == "<":
|
||||
garbage += parse_garbage(stream)
|
||||
else:
|
||||
raise Exception("Incorrectly formatted input")
|
||||
|
||||
# Determine whether to stop parsing
|
||||
if stream.peek() == "}":
|
||||
break
|
||||
elif stream.peek() == ",":
|
||||
assert stream.pop() == ","
|
||||
else:
|
||||
raise Exception("Incorrectly formatted input")
|
||||
|
||||
assert stream.pop() == "}"
|
||||
return groups, garbage
|
||||
|
||||
def parse_garbage(stream):
|
||||
assert stream.pop() == "<"
|
||||
|
||||
escaped = False
|
||||
garbage = 0
|
||||
|
||||
while True:
|
||||
if escaped:
|
||||
stream.pop()
|
||||
escaped = False
|
||||
elif stream.peek() == "!":
|
||||
assert stream.pop() == "!"
|
||||
escaped = True
|
||||
elif stream.peek() == ">":
|
||||
break
|
||||
else:
|
||||
stream.pop()
|
||||
garbage += 1
|
||||
|
||||
assert stream.pop() == ">"
|
||||
|
||||
return garbage
|
||||
|
||||
def group_score(group, level=1):
|
||||
return sum(group_score(subgroup, level + 1) for subgroup in group) + level
|
||||
|
||||
# PART 2
|
||||
|
||||
def main(filename):
|
||||
print(f"Solutions for {filename}")
|
||||
groups = load_groups(filename)
|
||||
for group, garbage in groups:
|
||||
score = group_score(group)
|
||||
print(f"Part 1: {score}")
|
||||
print(f"Part 2: {garbage}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
for filename in sys.argv[1:]:
|
||||
main(filename)
|
||||
Loading…
Add table
Add a link
Reference in a new issue