From 67130ccc370e9e6fb5f23a1a0f702cbd0f3a1c55 Mon Sep 17 00:00:00 2001 From: pegasust Date: Mon, 5 Dec 2022 05:44:56 +0000 Subject: [PATCH] d5: python --- 2022/d5/.envrc | 4 + 2022/d5/data/example.txt | 10 + 2022/d5/data/submission.txt | 513 ++++++++++++++++++++++++++++++++++++ 2022/d5/flake.lock | 43 +++ 2022/d5/flake.nix | 50 ++++ 2022/d5/run-py.sh | 7 + 2022/d5/src/d5.py | 77 ++++++ 7 files changed, 704 insertions(+) create mode 100644 2022/d5/.envrc create mode 100644 2022/d5/data/example.txt create mode 100644 2022/d5/data/submission.txt create mode 100644 2022/d5/flake.lock create mode 100644 2022/d5/flake.nix create mode 100755 2022/d5/run-py.sh create mode 100644 2022/d5/src/d5.py diff --git a/2022/d5/.envrc b/2022/d5/.envrc new file mode 100644 index 0000000..8ba04f9 --- /dev/null +++ b/2022/d5/.envrc @@ -0,0 +1,4 @@ +if command -v nix-shell &> /dev/null + then + use flake +fi diff --git a/2022/d5/data/example.txt b/2022/d5/data/example.txt new file mode 100644 index 0000000..192954e --- /dev/null +++ b/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 + diff --git a/2022/d5/data/submission.txt b/2022/d5/data/submission.txt new file mode 100644 index 0000000..19a9f09 --- /dev/null +++ b/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 + diff --git a/2022/d5/flake.lock b/2022/d5/flake.lock new file mode 100644 index 0000000..a62a638 --- /dev/null +++ b/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 +} diff --git a/2022/d5/flake.nix b/2022/d5/flake.nix new file mode 100644 index 0000000..e986fc0 --- /dev/null +++ b/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; + + }; + }; + } + ); +} diff --git a/2022/d5/run-py.sh b/2022/d5/run-py.sh new file mode 100755 index 0000000..1fc16f1 --- /dev/null +++ b/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 + diff --git a/2022/d5/src/d5.py b/2022/d5/src/d5.py new file mode 100644 index 0000000..90f6df9 --- /dev/null +++ b/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)