aboutsummaryrefslogtreecommitdiff
path: root/2021/10/puzzles.py
blob: 0554d25651ce6404980abf5060018f9fa795438f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#!/usr/bin/env python3

# START PART 1
from collections import Counter
# END PART 1


def main() -> None:
	with open("input", "r", encoding="utf-8") as f:
		data = list(map(lambda l: l.strip(), f.readlines()))

	# START PART 1
	counts = Counter()
	# END PART 1 START PART 2
	scores = []
	# END PART 2

	for line in data:
		stack = []
		for char in line:
			if char in ("(", "[", "{", "<"):
				stack.append(char)
			elif (stack.pop(), char) not in (("(", ")"), ("[", "]"), ("{", "}"), ("<", ">")):
				# START PART 1
				counts[char] += 1
				# END PART 1
				break
		# START PART 2
		else:
			score = 0
			for char in reversed(stack):
				score *= 5
				if char == "(":
					score += 1
				elif char == "[":
					score += 2
				elif char == "{":
					score += 3
				elif char == "<":
					score += 4
			scores.append(score)
		# END PART 2

	# START PART 1
	print(counts[")"] * 3 + counts["]"] * 57 + counts["}"] * 1197 + counts[">"] * 25137)
	# END PART 1 START PART 2
	scores.sort()
	print(scores[len(scores) // 2])
	# END PART 2


if __name__ == "__main__":
	main()