Update file names and init.shHEADmain

1 files changed, 51 insertions, 0 deletions

diff --git a/day02/solution.nim b/day02/solution.nim
new file mode 100644
index 0000000..8a794d3
--- /dev/null
+++ b/day02/solution.nim
@@ -0,0 +1,51 @@
+import std/strutils
+import std/sequtils
+
+let content = readFile("./input.txt").strip().split("\n")
+let rounds = map(
+    content,
+    # Split line into pair of characters
+    proc(round: string): tuple[other: char, self: char] = 
+       var splitted = round.split(" ")
+       (splitted[0][0], splitted[1][0]))
+
+# example: rounds = [('A', 'Z'), ('C', 'Y')]
+# echo rounds
+
+proc solve(dScore: array[3, array[3, int]]): int = 
+    var score = 0
+    for round in rounds:
+        let
+            i = ord(round.other) - ord('A')
+            j = ord(round.self) - ord('X')
+        score += dScore[i][j]
+
+    return score
+
+let pt1 = [
+    # A = Rock
+    # X Y Z
+    [1 + 3, 2 + 6, 3 + 0],
+    # B = Paper
+    # X Y Z
+    [1 + 0, 2 + 3, 3 + 6],
+    # C = Scissors
+    # X Y Z
+    [1 + 6, 2 + 0, 3 + 3],
+]
+
+echo solve(pt1)
+
+let pt2 = [
+    # A = Rock
+    # X Y Z
+    [3 + 0, 1 + 3, 2 + 6],
+    # B = Paper
+    # X Y Z
+    [1 + 0, 2 + 3, 3 + 6],
+    # C = Scissors
+    # X Y Z
+    [2 + 0, 3 + 3, 1 + 6],
+]
+
+echo solve(pt2)