d5: python

2022/d5/.envrc

@@ -0,0 +1,4 @@
+if command -v nix-shell &> /dev/null
+    then
+    use flake
+fi

2022/d5/data/example.txt

@@ -0,0 +1,10 @@
+[D]    
+[N] [C]    
+[Z] [M] [P]
+ 1   2   3 
+
+ move 1 from 2 to 1
+ move 3 from 1 to 3
+ move 2 from 2 to 1
+ move 1 from 1 to 2
+

2022/d5/data/submission.txt

@@ -0,0 +1,513 @@
+[P]     [L]         [T]            
+[L]     [M] [G]     [G]     [S]    
+[M]     [Q] [W]     [H] [R] [G]    
+[N]     [F] [M]     [D] [V] [R] [N]
+[W]     [G] [Q] [P] [J] [F] [M] [C]
+[V] [H] [B] [F] [H] [M] [B] [H] [B]
+[B] [Q] [D] [T] [T] [B] [N] [L] [D]
+[H] [M] [N] [Z] [M] [C] [M] [P] [P]
+ 1   2   3   4   5   6   7   8   9 
+
+ move 8 from 3 to 2
+ move 1 from 9 to 5
+ move 5 from 4 to 7
+ move 6 from 1 to 4
+ move 8 from 6 to 8
+ move 8 from 4 to 5
+ move 4 from 9 to 5
+ move 4 from 7 to 9
+ move 7 from 7 to 2
+ move 4 from 5 to 2
+ move 11 from 8 to 3
+ move 3 from 9 to 7
+ move 11 from 2 to 8
+ move 13 from 8 to 4
+ move 11 from 5 to 6
+ move 8 from 2 to 4
+ move 1 from 5 to 4
+ move 1 from 3 to 2
+ move 2 from 2 to 1
+ move 2 from 8 to 5
+ move 3 from 7 to 5
+ move 1 from 4 to 7
+ move 9 from 6 to 7
+ move 1 from 6 to 5
+ move 1 from 1 to 4
+ move 3 from 1 to 9
+ move 15 from 4 to 3
+ move 2 from 4 to 1
+ move 1 from 1 to 9
+ move 3 from 4 to 5
+ move 1 from 4 to 1
+ move 1 from 7 to 2
+ move 1 from 6 to 3
+ move 5 from 7 to 1
+ move 19 from 3 to 9
+ move 7 from 1 to 2
+ move 24 from 9 to 7
+ move 23 from 7 to 1
+ move 1 from 4 to 6
+ move 3 from 7 to 3
+ move 1 from 6 to 1
+ move 6 from 2 to 1
+ move 21 from 1 to 9
+ move 5 from 3 to 8
+ move 2 from 2 to 5
+ move 10 from 9 to 5
+ move 1 from 2 to 1
+ move 5 from 1 to 3
+ move 6 from 3 to 4
+ move 1 from 2 to 8
+ move 3 from 5 to 2
+ move 4 from 9 to 3
+ move 13 from 5 to 9
+ move 2 from 7 to 2
+ move 3 from 4 to 7
+ move 1 from 7 to 8
+ move 5 from 1 to 3
+ move 1 from 7 to 5
+ move 1 from 8 to 1
+ move 2 from 2 to 7
+ move 19 from 9 to 2
+ move 5 from 2 to 3
+ move 7 from 5 to 9
+ move 1 from 1 to 9
+ move 5 from 9 to 2
+ move 4 from 9 to 3
+ move 20 from 3 to 9
+ move 1 from 3 to 9
+ move 3 from 7 to 3
+ move 16 from 2 to 3
+ move 12 from 3 to 4
+ move 2 from 2 to 5
+ move 1 from 2 to 4
+ move 2 from 4 to 1
+ move 4 from 8 to 1
+ move 15 from 9 to 3
+ move 2 from 5 to 3
+ move 3 from 2 to 8
+ move 5 from 8 to 5
+ move 7 from 3 to 4
+ move 2 from 9 to 6
+ move 15 from 3 to 1
+ move 3 from 1 to 8
+ move 3 from 9 to 5
+ move 9 from 4 to 1
+ move 3 from 3 to 5
+ move 2 from 6 to 5
+ move 9 from 1 to 3
+ move 1 from 9 to 4
+ move 1 from 5 to 2
+ move 3 from 8 to 5
+ move 10 from 1 to 6
+ move 12 from 4 to 8
+ move 1 from 2 to 7
+ move 2 from 5 to 6
+ move 1 from 1 to 4
+ move 7 from 3 to 6
+ move 1 from 7 to 2
+ move 2 from 4 to 9
+ move 3 from 1 to 7
+ move 1 from 9 to 8
+ move 1 from 2 to 3
+ move 3 from 1 to 7
+ move 5 from 8 to 2
+ move 5 from 7 to 1
+ move 9 from 6 to 8
+ move 6 from 6 to 9
+ move 8 from 8 to 6
+ move 1 from 7 to 4
+ move 5 from 2 to 4
+ move 7 from 5 to 1
+ move 5 from 8 to 9
+ move 11 from 6 to 7
+ move 9 from 9 to 1
+ move 2 from 7 to 5
+ move 1 from 9 to 5
+ move 1 from 3 to 6
+ move 3 from 4 to 6
+ move 1 from 8 to 2
+ move 2 from 3 to 6
+ move 6 from 5 to 2
+ move 3 from 5 to 9
+ move 3 from 2 to 1
+ move 1 from 4 to 3
+ move 3 from 2 to 7
+ move 1 from 8 to 9
+ move 1 from 2 to 8
+ move 8 from 7 to 5
+ move 1 from 7 to 8
+ move 3 from 5 to 6
+ move 5 from 5 to 2
+ move 1 from 4 to 1
+ move 1 from 3 to 2
+ move 4 from 1 to 5
+ move 4 from 2 to 6
+ move 6 from 1 to 2
+ move 5 from 9 to 3
+ move 2 from 5 to 3
+ move 3 from 3 to 6
+ move 10 from 6 to 4
+ move 4 from 8 to 5
+ move 5 from 5 to 1
+ move 21 from 1 to 7
+ move 3 from 2 to 9
+ move 1 from 5 to 2
+ move 4 from 2 to 9
+ move 8 from 4 to 8
+ move 1 from 2 to 1
+ move 7 from 8 to 2
+ move 2 from 6 to 1
+ move 2 from 1 to 5
+ move 1 from 1 to 5
+ move 4 from 3 to 7
+ move 1 from 9 to 3
+ move 4 from 6 to 3
+ move 1 from 3 to 8
+ move 1 from 3 to 4
+ move 2 from 2 to 6
+ move 2 from 9 to 7
+ move 14 from 7 to 8
+ move 10 from 8 to 7
+ move 3 from 4 to 6
+ move 5 from 2 to 3
+ move 3 from 9 to 8
+ move 3 from 3 to 4
+ move 1 from 2 to 4
+ move 1 from 9 to 4
+ move 1 from 9 to 5
+ move 1 from 5 to 2
+ move 3 from 5 to 7
+ move 1 from 4 to 6
+ move 5 from 3 to 8
+ move 1 from 6 to 8
+ move 5 from 7 to 6
+ move 14 from 8 to 5
+ move 2 from 6 to 7
+ move 18 from 7 to 2
+ move 3 from 6 to 1
+ move 5 from 5 to 4
+ move 5 from 6 to 2
+ move 7 from 2 to 1
+ move 1 from 8 to 4
+ move 1 from 5 to 1
+ move 8 from 1 to 9
+ move 10 from 4 to 3
+ move 8 from 5 to 3
+ move 1 from 4 to 3
+ move 2 from 1 to 5
+ move 1 from 5 to 3
+ move 5 from 3 to 1
+ move 1 from 1 to 3
+ move 5 from 1 to 6
+ move 13 from 3 to 1
+ move 3 from 9 to 4
+ move 2 from 9 to 6
+ move 5 from 6 to 5
+ move 6 from 5 to 1
+ move 7 from 7 to 9
+ move 7 from 9 to 6
+ move 1 from 9 to 3
+ move 1 from 7 to 9
+ move 3 from 9 to 1
+ move 12 from 2 to 7
+ move 7 from 6 to 2
+ move 22 from 1 to 7
+ move 1 from 6 to 5
+ move 4 from 7 to 6
+ move 1 from 5 to 6
+ move 2 from 4 to 1
+ move 1 from 4 to 1
+ move 23 from 7 to 9
+ move 4 from 6 to 2
+ move 4 from 7 to 3
+ move 1 from 1 to 9
+ move 6 from 2 to 1
+ move 1 from 7 to 2
+ move 7 from 2 to 8
+ move 2 from 3 to 8
+ move 3 from 1 to 9
+ move 1 from 2 to 8
+ move 5 from 8 to 3
+ move 3 from 2 to 1
+ move 2 from 7 to 8
+ move 10 from 9 to 8
+ move 4 from 1 to 3
+ move 14 from 3 to 4
+ move 7 from 4 to 5
+ move 1 from 6 to 9
+ move 5 from 5 to 8
+ move 1 from 6 to 4
+ move 6 from 9 to 4
+ move 3 from 8 to 4
+ move 1 from 5 to 1
+ move 3 from 4 to 3
+ move 9 from 4 to 3
+ move 5 from 3 to 6
+ move 5 from 1 to 5
+ move 4 from 6 to 2
+ move 8 from 9 to 2
+ move 2 from 6 to 5
+ move 3 from 4 to 7
+ move 2 from 2 to 7
+ move 2 from 5 to 4
+ move 3 from 5 to 9
+ move 3 from 4 to 2
+ move 10 from 2 to 5
+ move 1 from 9 to 8
+ move 2 from 2 to 9
+ move 3 from 7 to 2
+ move 1 from 2 to 9
+ move 13 from 5 to 1
+ move 2 from 2 to 7
+ move 8 from 9 to 2
+ move 1 from 4 to 6
+ move 1 from 9 to 5
+ move 14 from 8 to 4
+ move 7 from 4 to 5
+ move 4 from 7 to 5
+ move 2 from 3 to 8
+ move 4 from 1 to 5
+ move 2 from 5 to 4
+ move 6 from 5 to 6
+ move 7 from 2 to 5
+ move 1 from 2 to 6
+ move 1 from 5 to 2
+ move 2 from 2 to 8
+ move 2 from 1 to 3
+ move 8 from 4 to 7
+ move 1 from 4 to 3
+ move 6 from 1 to 6
+ move 7 from 3 to 9
+ move 3 from 7 to 1
+ move 2 from 8 to 7
+ move 7 from 6 to 9
+ move 2 from 3 to 6
+ move 6 from 8 to 3
+ move 9 from 5 to 3
+ move 2 from 7 to 8
+ move 2 from 6 to 4
+ move 7 from 6 to 9
+ move 5 from 3 to 8
+ move 10 from 9 to 1
+ move 11 from 1 to 8
+ move 1 from 3 to 2
+ move 4 from 5 to 6
+ move 2 from 6 to 2
+ move 2 from 7 to 9
+ move 3 from 1 to 7
+ move 6 from 3 to 9
+ move 2 from 7 to 2
+ move 2 from 6 to 9
+ move 1 from 5 to 9
+ move 11 from 9 to 8
+ move 1 from 4 to 5
+ move 6 from 9 to 8
+ move 31 from 8 to 9
+ move 1 from 3 to 6
+ move 1 from 7 to 1
+ move 1 from 4 to 3
+ move 1 from 5 to 2
+ move 1 from 1 to 8
+ move 1 from 8 to 9
+ move 1 from 7 to 3
+ move 11 from 9 to 6
+ move 2 from 3 to 1
+ move 2 from 3 to 5
+ move 1 from 5 to 4
+ move 1 from 4 to 1
+ move 6 from 8 to 3
+ move 1 from 1 to 4
+ move 1 from 4 to 6
+ move 2 from 3 to 6
+ move 17 from 9 to 2
+ move 23 from 2 to 9
+ move 14 from 9 to 4
+ move 1 from 1 to 7
+ move 1 from 5 to 6
+ move 8 from 6 to 2
+ move 1 from 3 to 2
+ move 4 from 9 to 8
+ move 5 from 4 to 7
+ move 3 from 7 to 2
+ move 1 from 1 to 2
+ move 2 from 9 to 4
+ move 3 from 6 to 9
+ move 8 from 4 to 9
+ move 2 from 4 to 2
+ move 4 from 7 to 2
+ move 1 from 7 to 9
+ move 4 from 6 to 2
+ move 16 from 2 to 1
+ move 2 from 3 to 2
+ move 18 from 9 to 8
+ move 1 from 4 to 2
+ move 1 from 6 to 8
+ move 1 from 3 to 9
+ move 3 from 9 to 5
+ move 4 from 9 to 8
+ move 6 from 2 to 8
+ move 1 from 5 to 1
+ move 4 from 2 to 8
+ move 1 from 5 to 1
+ move 17 from 1 to 4
+ move 1 from 5 to 8
+ move 10 from 4 to 3
+ move 10 from 3 to 1
+ move 4 from 4 to 9
+ move 1 from 4 to 6
+ move 1 from 4 to 8
+ move 38 from 8 to 1
+ move 27 from 1 to 5
+ move 1 from 8 to 2
+ move 1 from 6 to 3
+ move 1 from 4 to 8
+ move 1 from 8 to 4
+ move 14 from 1 to 9
+ move 1 from 3 to 1
+ move 1 from 5 to 1
+ move 1 from 2 to 5
+ move 2 from 5 to 4
+ move 17 from 5 to 8
+ move 3 from 4 to 9
+ move 2 from 9 to 1
+ move 3 from 5 to 7
+ move 3 from 7 to 4
+ move 2 from 4 to 7
+ move 12 from 1 to 4
+ move 1 from 7 to 4
+ move 1 from 7 to 6
+ move 1 from 6 to 9
+ move 11 from 4 to 3
+ move 1 from 5 to 3
+ move 11 from 3 to 9
+ move 1 from 3 to 2
+ move 3 from 5 to 4
+ move 1 from 2 to 4
+ move 1 from 5 to 8
+ move 13 from 9 to 3
+ move 16 from 9 to 1
+ move 4 from 8 to 9
+ move 2 from 1 to 4
+ move 1 from 9 to 1
+ move 1 from 9 to 7
+ move 1 from 7 to 2
+ move 6 from 8 to 3
+ move 8 from 4 to 2
+ move 4 from 9 to 6
+ move 3 from 2 to 3
+ move 3 from 6 to 1
+ move 3 from 8 to 6
+ move 1 from 6 to 8
+ move 3 from 6 to 4
+ move 11 from 3 to 5
+ move 4 from 8 to 2
+ move 6 from 3 to 5
+ move 3 from 5 to 1
+ move 2 from 8 to 3
+ move 14 from 5 to 3
+ move 4 from 3 to 4
+ move 6 from 3 to 5
+ move 3 from 2 to 9
+ move 4 from 1 to 8
+ move 3 from 9 to 6
+ move 2 from 6 to 9
+ move 6 from 4 to 3
+ move 15 from 1 to 4
+ move 1 from 6 to 7
+ move 5 from 5 to 1
+ move 11 from 3 to 1
+ move 2 from 9 to 7
+ move 1 from 5 to 6
+ move 2 from 1 to 3
+ move 7 from 2 to 6
+ move 4 from 8 to 1
+ move 8 from 4 to 2
+ move 3 from 6 to 4
+ move 5 from 1 to 4
+ move 17 from 4 to 8
+ move 3 from 3 to 7
+ move 4 from 3 to 4
+ move 4 from 4 to 2
+ move 9 from 8 to 7
+ move 1 from 3 to 8
+ move 10 from 2 to 4
+ move 1 from 6 to 2
+ move 2 from 8 to 4
+ move 2 from 6 to 9
+ move 2 from 6 to 2
+ move 1 from 2 to 3
+ move 3 from 1 to 4
+ move 1 from 3 to 2
+ move 1 from 9 to 3
+ move 1 from 9 to 7
+ move 4 from 8 to 4
+ move 10 from 4 to 8
+ move 5 from 4 to 3
+ move 1 from 2 to 8
+ move 5 from 3 to 7
+ move 3 from 7 to 8
+ move 3 from 4 to 3
+ move 8 from 7 to 2
+ move 8 from 7 to 8
+ move 1 from 3 to 2
+ move 3 from 2 to 8
+ move 9 from 2 to 5
+ move 12 from 1 to 7
+ move 21 from 8 to 3
+ move 5 from 8 to 6
+ move 8 from 7 to 5
+ move 6 from 7 to 4
+ move 12 from 5 to 7
+ move 1 from 8 to 5
+ move 2 from 4 to 2
+ move 1 from 7 to 6
+ move 14 from 3 to 8
+ move 5 from 6 to 2
+ move 7 from 2 to 6
+ move 6 from 8 to 4
+ move 11 from 7 to 4
+ move 8 from 3 to 7
+ move 4 from 5 to 7
+ move 9 from 8 to 2
+ move 6 from 4 to 1
+ move 2 from 5 to 2
+ move 1 from 7 to 2
+ move 11 from 2 to 3
+ move 1 from 2 to 1
+ move 7 from 4 to 1
+ move 5 from 6 to 8
+ move 1 from 2 to 3
+ move 2 from 8 to 7
+ move 14 from 3 to 7
+ move 15 from 7 to 6
+ move 4 from 4 to 6
+ move 2 from 8 to 3
+ move 12 from 1 to 3
+ move 1 from 8 to 2
+ move 1 from 2 to 3
+ move 1 from 3 to 9
+ move 1 from 9 to 7
+ move 1 from 1 to 4
+ move 18 from 6 to 8
+ move 3 from 3 to 2
+ move 17 from 8 to 3
+ move 3 from 7 to 6
+ move 3 from 2 to 6
+ move 25 from 3 to 7
+ move 2 from 4 to 1
+ move 9 from 6 to 5
+ move 2 from 3 to 1
+ move 1 from 3 to 9
+ move 5 from 5 to 2
+ move 1 from 8 to 3
+ move 2 from 4 to 7
+ move 1 from 9 to 4
+ move 1 from 6 to 7
+ move 2 from 5 to 2
+ move 2 from 4 to 8
+ move 2 from 5 to 8
+ move 5 from 7 to 9
+ move 27 from 7 to 5
+ move 2 from 9 to 6
+

2022/d5/flake.lock

@@ -0,0 +1,43 @@
+{
+  "nodes": {
+    "flake-utils": {
+      "locked": {
+        "lastModified": 1667395993,
+        "narHash": "sha256-nuEHfE/LcWyuSWnS8t12N1wc105Qtau+/OdUAjtQ0rA=",
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "rev": "5aed5285a952e0b949eb3ba02c12fa4fcfef535f",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "type": "github"
+      }
+    },
+    "nixpkgs": {
+      "locked": {
+        "lastModified": 1670064435,
+        "narHash": "sha256-+ELoY30UN+Pl3Yn7RWRPabykwebsVK/kYE9JsIsUMxQ=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "61a8a98e6d557e6dd7ed0cdb54c3a3e3bbc5e25c",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "ref": "nixos-unstable",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "root": {
+      "inputs": {
+        "flake-utils": "flake-utils",
+        "nixpkgs": "nixpkgs"
+      }
+    }
+  },
+  "root": "root",
+  "version": 7
+}

2022/d5/flake.nix

@@ -0,0 +1,50 @@
+{
+  description = "D4 AOC with Lua!";
+  inputs = {
+    nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
+    flake-utils.url = "github:numtide/flake-utils";
+  };
+  outputs = { nixpkgs, flake-utils, ... } @ inputs:
+    flake-utils.lib.eachSystem flake-utils.lib.defaultSystems (sys:
+      let
+        overlays = [ ];
+        pkgs = import nixpkgs { system = sys; overlays = overlays; };
+        shellHookAfter = ''
+          echo "The input files should be placed under ./data/{submission,example}.txt"
+          echo "This problem shares one input between two parts"
+        '';
+        py_pkgs = [ pkgs.python310 ];
+        lua_pkgs = [ (pkgs.lua.withPackages (luapkgs: [ luapkgs.busted luapkgs.luafilesystem ])) ];
+        fennel_pkgs = [ pkgs.lua.withPackages (luapkgs: [ luapkgs.fennel ]) ];
+      in
+      {
+        # Jack of all trades
+        devShell = pkgs.mkShell {
+          nativeBuildInputs = py_pkgs ++ fennel_pkgs;
+          shellHook = ''
+            echo "> Default runtime. This contains both lua and python3 env"
+            echo "Run ./run-py.sh for Python's output and ./run-lua.sh for Lua's output"
+          '' + shellHookAfter;
+        };
+        devShells = {
+          # nix develop ./#lua
+          # lua = pkgs.mkShell {
+          #   nativeBuildInputs = lua_pkgs;
+          #   shellHook = ''
+          #     echo "> Lua runtime"
+          #     echo "Run ./run-lua.sh to see the output of the solution"
+          #   '' + shellHookAfter;
+          # };
+          # nix develop ./#python
+          python = pkgs.mkShell {
+            nativeBuildInputs = py_pkgs;
+            shellHook = ''
+              echo "> Python3 runtime"
+              echo "Run ./run-py.sh to see the output of the solution"
+            '' + shellHookAfter;
+
+          };
+        };
+      }
+    );
+}

2022/d5/run-py.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/env sh
+echo "example"
+python3 ./src/d5.py ./data/example.txt
+
+echo "submission"
+python3 ./src/d5.py ./data/submission.txt
+

2022/d5/src/d5.py

@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+from typing import Literal, Union, Iterable, Optional
+from dataclasses import dataclass
+
+@dataclass
+class MoveInsn:
+    how_many: int
+    from_crate: int
+    to_crate: int
+
+    @classmethod
+    def from_line(cls, line: str):
+        tokens = line.strip().split()
+        v = map(lambda v: int(tokens[v]), [1, 3, 5]) # indices of "number" token
+        return cls(*v) # this is basically calls the constructor
+
+def part1(crates: list[list[str]], insns: list[MoveInsn]) -> str:
+    for move in insns:
+        for _ in range(move.how_many):
+            crates[move.to_crate - 1].append(crates[move.from_crate - 1].pop())
+    return "".join([crate[-1] for crate in crates])
+
+def part2(crates: list[list[str]], insns: list[MoveInsn]) -> str:
+    for move in insns:
+        pops =([crates[move.from_crate - 1].pop() for _ in range(move.how_many)])
+        crates[move.to_crate - 1].extend(reversed(pops))
+    return "".join([crate[-1] for crate in crates])
+
+
+def copy_crates(crate_stacks: list[list[str]]):
+    return [c[:] for c in crate_stacks]
+
+def main(input_lines: Iterable[str]):
+    # Assuming each crate name is single char
+    PARSE_CRATE = 0
+    PARSE_INSNS = 1
+
+    parse_state = 0
+    crates_str = []
+    crate_stacks: Optional[list[list[str]]] = None
+    moves = []
+
+    for _line in input_lines:
+        if len(_line.strip()) == 0:
+            # neovim copy-paste doesn't work too well, excuse me
+            continue
+        match parse_state:
+            case 0:
+                crates = [s.strip() for s in _line.split()]
+                if crates[0] == "1":
+                    parse_state = PARSE_INSNS
+                    # Begin propagating
+                    crate_stacks = [[] for _ in range(len(crates))]
+                    for crate_line in reversed(crates_str):
+                        crate_decls = [crate_line[i] for i in range(1, len(crate_line), 4)]
+                        for crate, stack in zip(crate_decls, crate_stacks):
+                            if crate == ' ':
+                                continue
+                            stack.append(crate)
+                    # print(crate_stacks)
+                else:
+                    crates_str.append(_line)
+                pass
+            case 1:
+                moves.append(MoveInsn.from_line(_line))
+                # print(moves)
+                pass
+
+    assert crate_stacks is not None, "input must indicate crates input"
+    print("part1", part1(copy_crates(crate_stacks), moves))
+    print("part2", part2(copy_crates(crate_stacks), moves))
+    pass
+
+if __name__ == "__main__":
+    import sys
+    with open(sys.argv[1], "r") as f:
+        main(f)