Added 2019 day 03

2019/03/code.py

@@ -0,0 +1,76 @@
+""" https://adventofcode.com/2019/day/3 """
+
+def readFile():
+    with open(f"{__file__.rstrip('code.py')}input.txt", "r") as f:
+        return [Wire(val for val in line[:-1].split(",")) for line in f.readlines()]
+
+class Wire:
+    def __init__(self, input):
+        self.input = input
+        self.fields = {(0, 0, 0)}
+        self.__initFields()
+
+    def __initFields(self):
+        x, y, step = 0, 0, 0
+        for i in self.input:
+            dir = i[0]
+            len = int(i[1:])
+            dx = 1 if dir == "R" else -1 if dir == "L" else 0
+            dy = 1 if dir == "U" else -1 if dir == "D" else 0
+            for r in range(0, len):
+                x += dx
+                y += dy
+                step += 1
+                self.fields.add((x, y, step))
+        
+def getIntersections(a , b):
+    intersections = set()
+    bFields = {(val[0], val[1]) for val in b.fields}
+    for point in a.fields:
+        if point[:2] in bFields:
+            intersections.add(point)
+    return intersections
+
+def getIntersectionDistance(intersections, i):
+    for intersection in intersections:
+        if intersection[0] == i[0] and intersection[1] == i[1]:
+            return intersection[2]
+                
+def part1(vals : list):
+    intersections = getIntersections(vals[0], vals[1])
+    dists = []
+    for i in intersections:
+        dists.append(abs(i[0]) + abs(i[1])) # Manhattan Distance
+    dists.sort()
+    return dists[1]
+
+def part2(vals : list):
+    intersections = getIntersections(vals[0], vals[1])
+    intersections2 = getIntersections(vals[1], vals[0])
+    steps = []
+    for i in intersections:
+        steps.append(i[2] + getIntersectionDistance(intersections2, i))
+    steps.sort()
+    return steps[1]
+
+def test():
+    w1 = Wire([val for val in "R8,U5,L5,D3".split(",")])
+    w2 = Wire([val for val in "U7,R6,D4,L4".split(",")])
+    assert(part1([w1, w2]) == 6), "Failed 1 - 1"
+    assert(part2([w1, w2]) == 30), "Failed 1 - 2"
+
+    w3 = Wire([val for val in "R75,D30,R83,U83,L12,D49,R71,U7,L72".split(",")])
+    w4 = Wire([val for val in "U62,R66,U55,R34,D71,R55,D58,R83".split(",")])
+    assert(part1([w3, w4]) == 159), "Failed 2 - 1"
+    assert(part2([w3, w4]) == 610), "Failed 2 - 2"
+
+    w5 = Wire([val for val in "R98,U47,R26,D63,R33,U87,L62,D20,R33,U53,R51".split(",")])
+    w6 = Wire([val for val in "U98,R91,D20,R16,D67,R40,U7,R15,U6,R7".split(",")])
+    assert(part1([w5, w6]) == 135), "Failed 3 - 1"
+    assert(part2([w5, w6]) == 410), "Failed 3 - 2"
+
+if __name__ == "__main__":
+    test()
+    vals = readFile()
+    print(f"Part 1: {part1(vals)}")
+    print(f"Part 2: {part2(vals)}")