rename stuff

2022/day02/code.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python3
+
+import sys
+import numpy as np
+
+def calc_score(game):
+    a = ['A', 'B', 'C'].index(game[0])
+    b = ['X', 'Y', 'Z'].index(game[1])
+    if a == b: return 1 + b + 3
+    elif b == ((a-1) % 3): return 1 + b
+    else: return 1 + b + 6
+
+def calc_moves(game):
+    a = ['A', 'B', 'C'].index(game[0])
+    o = ['X', 'Y', 'Z'].index(game[1])
+    if o == 1: return 1 + a + 3
+    elif o == 0: return 1 + ((a - 1) % 3)
+    else: return 1 + ((a + 1) % 3) + 6
+
+if __name__ == '__main__':
+    games = [tuple(line.strip('\n').split(' ')) for line in open(sys.argv[1])]
+
+    # challenge 1
+    scores = list(map(calc_score,games))
+    res1 = str(np.sum(scores))
+    print("challenge 1:" + "\n" +  res1 + "\n")
+
+    # challenge 2
+    scores = list(map(calc_moves,games))
+    res2 = str(np.sum(scores))
+    print("challenge 2:" + "\n" +  res2 + "\n")
+