Update 2015 solutions

2015/d3.py

@@ -1,9 +1,7 @@
-import sys
-from lib import *
+from lib import get_data
+from lib import V
 
-data = open(0).read()
-
-part_1 = False
+data = get_data(__file__)
 
 DIRS = {
     "^": (-1, 0),
@@ -12,30 +10,27 @@ DIRS = {
     "<": (0, -1),
 }
 
-if part_1:
-    pos = (0, 0)
-    poss = set([pos])
+pos = V(0, 0)
+poss = set([pos])
 
-    for c in data:
-        d = DIRS[c]
-        pos = pos[0] + d[0], pos[1] + d[1]
-        poss.add(pos)
+for c in data:
+    d = DIRS[c]
+    pos = pos + d
+    poss.add(pos)
 
-    print(len(poss))
-    sys.exit(0)
+print(len(poss))
 
-a = (0, 0)
-b = (0, 0)
+a = V(0, 0)
+b = V(0, 0)
 poss = set([a, b])
 for i, c in enumerate(data):
     if i % 2 == 0:
         d = DIRS[c]
-        a = a[0] + d[0], a[1] + d[1]
+        a = a + d
         poss.add(a)
     else:
         d = DIRS[c]
-        b = b[0] + d[0], b[1] + d[1]
+        b = b + d
         poss.add(b)
 
 print(len(poss))
-sys.exit(0)