47 lines
906 B
Python
47 lines
906 B
Python
import sys
|
|
|
|
def load_steps(filename):
|
|
with open(filename, "r") as f:
|
|
return f.read()[:-1]
|
|
|
|
# PART 1
|
|
|
|
def find_doubles(steps):
|
|
x, y = 0, 0
|
|
houses = {(0, 0)}
|
|
for step in steps:
|
|
if step == "^": y += 1
|
|
elif step == "v": y -= 1
|
|
elif step == "<": x -= 1
|
|
elif step == ">": x += 1
|
|
houses.add((x, y))
|
|
return houses
|
|
|
|
# PART 2
|
|
|
|
def split_string(s):
|
|
s1 = ""
|
|
s2 = ""
|
|
while True:
|
|
if not s: break
|
|
s1 += s[0]
|
|
s = s[1:]
|
|
if not s: break
|
|
s2 += s[0]
|
|
s = s[1:]
|
|
return s1, s2
|
|
|
|
def main(filename):
|
|
steps = load_steps(filename)
|
|
print(f"Solutions for {filename}")
|
|
doubles = len(find_doubles(steps))
|
|
print(f"Part 1: {doubles}")
|
|
santa, robot = split_string(steps)
|
|
santa_doubles = find_doubles(santa)
|
|
robot_doubles = find_doubles(robot)
|
|
doubles_2 = len(santa_doubles | robot_doubles)
|
|
print(f"Part 2: {doubles_2}")
|
|
|
|
if __name__ == "__main__":
|
|
for filename in sys.argv[1:]:
|
|
main(filename)
|