add day5

3 years ago · d287410094
parent 764c2e4942
commit d287410094
3 changed files with 617 additions and 0 deletions
--- a/day5/day5.jl
+++ b/day5/day5.jl
@ -0,0 +1,97 @@
+#!/usr/bin/env julia
+
+using Test
+
+function parse_stacks_line(line)
+  parts = Iterators.partition(line, 4)
+  return [s[2] for s in parts]
+end
+
+function read_stacks(io)
+  stacks = Vector{Vector{Char}}()
+  for line in eachline(io)
+    if isempty(line)
+      return stacks
+    end
+    if line[1:3] == " 1 "
+      continue
+    end
+    stacks_row = parse_stacks_line(line)
+    while length(stacks_row) > length(stacks)
+      push!(stacks, [])
+    end
+    for (i, c) in Iterators.enumerate(stacks_row)
+      if c == ' '
+        continue
+      end
+      pushfirst!(stacks[i], c)
+    end
+  end
+  return stacks
+end
+
+struct Move
+  count :: Int
+  from :: Int
+  to :: Int
+end
+
+function parse_move_line(line)
+  # move 1 from 2 to 1
+  parts = split(line)
+  return Move(parse(Int, parts[2]), parse(Int, parts[4]), parse(Int, parts[6]))
+end
+
+function read_apply_moves(stacks, io; rev::Bool = true)
+  for line in eachline(io)
+    move = parse_move_line(line)
+    if rev
+      for i in 1:move.count
+        push!(stacks[move.to], pop!(stacks[move.from]))
+      end
+    else
+      from_len = length(stacks[move.from])
+      from_range = UnitRange(from_len - move.count + 1, from_len)
+      append!(stacks[move.to], splice!(stacks[move.from], from_range))
+    end
+  end
+  return stacks
+end
+
+function operate_crane(infile; rev::Bool = true)
+  open(infile, "r") do io
+    stacks = read_stacks(io)
+    stacks2 = read_apply_moves(stacks, io, rev=rev)
+    return stacks2
+  end
+end
+
+function top_crates(stacks)
+  return join([last(s) for s in stacks])
+end
+
+function test()
+  @testset "elf crates 2000" verbose=true begin
+    @test top_crates(operate_crane("example.txt")) == "CMZ"
+    @test top_crates(operate_crane("input.txt")) == "QNHWJVJZW"
+  end
+  @testset "elf crates 2001" verbose=true begin
+    @test top_crates(operate_crane("example.txt", rev=false)) == "MCD"
+    @test top_crates(operate_crane("input.txt", rev=false)) == "BPCZJLFJW"
+  end
+end
+
+function main()
+  if size(ARGS, 1) == 0
+    test()
+  else
+    infile = ARGS[1]
+    println("infile = ", infile)
+    println("crates: ", read_stacks(infile))
+    stacks2 = operate_crane(infile)
+    println("after : ", stacks2)
+    println("top   : ", top_crates(stacks2))
+  end
+end
+
+main()
--- a/day5/example.txt
+++ b/day5/example.txt
@ -0,0 +1,9 @@
+    [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
--- a/day5/input.txt
+++ b/day5/input.txt
@ -0,0 +1,511 @@
+[V]         [T]         [J]        
+[Q]         [M] [P]     [Q]     [J]
+[W] [B]     [N] [Q]     [C]     [T]
+[M] [C]     [F] [N]     [G] [W] [G]
+[B] [W] [J] [H] [L]     [R] [B] [C]
+[N] [R] [R] [W] [W] [W] [D] [N] [F]
+[Z] [Z] [Q] [S] [F] [P] [B] [Q] [L]
+[C] [H] [F] [Z] [G] [L] [V] [Z] [H]
+ 1   2   3   4   5   6   7   8   9 
+
+move 2 from 1 to 7
+move 6 from 2 to 6
+move 10 from 7 to 6
+move 4 from 3 to 1
+move 5 from 6 to 4
+move 1 from 1 to 9
+move 4 from 6 to 9
+move 12 from 4 to 1
+move 5 from 1 to 4
+move 7 from 9 to 8
+move 11 from 8 to 1
+move 6 from 6 to 2
+move 2 from 5 to 2
+move 3 from 6 to 3
+move 4 from 9 to 4
+move 2 from 2 to 5
+move 1 from 6 to 4
+move 3 from 3 to 6
+move 1 from 8 to 4
+move 1 from 6 to 1
+move 28 from 1 to 4
+move 28 from 4 to 5
+move 1 from 9 to 1
+move 4 from 4 to 1
+move 2 from 6 to 2
+move 2 from 1 to 6
+move 7 from 4 to 2
+move 14 from 2 to 9
+move 1 from 4 to 1
+move 1 from 1 to 2
+move 18 from 5 to 6
+move 2 from 2 to 6
+move 1 from 9 to 7
+move 8 from 9 to 2
+move 15 from 6 to 5
+move 1 from 6 to 3
+move 3 from 2 to 5
+move 1 from 7 to 5
+move 2 from 1 to 3
+move 3 from 2 to 1
+move 1 from 6 to 4
+move 5 from 6 to 5
+move 2 from 2 to 9
+move 35 from 5 to 7
+move 4 from 9 to 3
+move 1 from 4 to 1
+move 5 from 1 to 7
+move 6 from 5 to 3
+move 1 from 9 to 4
+move 11 from 7 to 6
+move 2 from 9 to 2
+move 1 from 4 to 7
+move 14 from 7 to 4
+move 5 from 6 to 9
+move 2 from 2 to 4
+move 6 from 7 to 9
+move 2 from 9 to 5
+move 6 from 9 to 5
+move 8 from 4 to 9
+move 5 from 4 to 3
+move 3 from 5 to 7
+move 1 from 3 to 9
+move 5 from 3 to 4
+move 7 from 9 to 8
+move 2 from 7 to 4
+move 4 from 5 to 7
+move 1 from 5 to 3
+move 5 from 6 to 4
+move 8 from 4 to 8
+move 5 from 7 to 6
+move 1 from 4 to 7
+move 3 from 6 to 9
+move 2 from 6 to 5
+move 7 from 8 to 3
+move 2 from 5 to 9
+move 17 from 3 to 1
+move 3 from 1 to 3
+move 6 from 8 to 9
+move 4 from 4 to 7
+move 6 from 3 to 5
+move 2 from 8 to 5
+move 14 from 7 to 5
+move 2 from 4 to 5
+move 6 from 9 to 5
+move 1 from 7 to 9
+move 1 from 6 to 9
+move 8 from 1 to 9
+move 8 from 5 to 2
+move 2 from 1 to 3
+move 7 from 2 to 6
+move 2 from 3 to 4
+move 1 from 2 to 6
+move 3 from 1 to 6
+move 16 from 9 to 4
+move 2 from 9 to 8
+move 1 from 1 to 6
+move 2 from 9 to 4
+move 1 from 6 to 9
+move 1 from 6 to 1
+move 1 from 1 to 7
+move 1 from 6 to 9
+move 1 from 9 to 3
+move 1 from 3 to 8
+move 1 from 9 to 2
+move 1 from 2 to 7
+move 2 from 5 to 3
+move 7 from 5 to 8
+move 2 from 7 to 9
+move 1 from 6 to 7
+move 3 from 6 to 9
+move 10 from 8 to 7
+move 1 from 4 to 3
+move 3 from 3 to 1
+move 1 from 7 to 1
+move 19 from 4 to 6
+move 3 from 9 to 7
+move 1 from 9 to 2
+move 2 from 1 to 7
+move 1 from 9 to 1
+move 12 from 6 to 9
+move 2 from 7 to 1
+move 1 from 2 to 4
+move 11 from 6 to 3
+move 1 from 4 to 8
+move 1 from 6 to 8
+move 11 from 7 to 9
+move 2 from 8 to 9
+move 18 from 9 to 6
+move 5 from 3 to 7
+move 5 from 3 to 8
+move 11 from 5 to 6
+move 26 from 6 to 4
+move 1 from 6 to 5
+move 1 from 3 to 7
+move 3 from 8 to 3
+move 1 from 8 to 7
+move 3 from 3 to 6
+move 5 from 9 to 3
+move 1 from 4 to 9
+move 8 from 4 to 5
+move 2 from 7 to 8
+move 3 from 3 to 6
+move 3 from 4 to 6
+move 7 from 7 to 4
+move 1 from 9 to 1
+move 5 from 5 to 3
+move 2 from 9 to 7
+move 3 from 8 to 2
+move 7 from 3 to 7
+move 1 from 7 to 6
+move 3 from 5 to 6
+move 7 from 4 to 8
+move 10 from 4 to 5
+move 2 from 4 to 2
+move 3 from 7 to 5
+move 2 from 4 to 1
+move 6 from 8 to 5
+move 5 from 1 to 4
+move 5 from 4 to 2
+move 5 from 7 to 8
+move 10 from 2 to 8
+move 3 from 8 to 3
+move 2 from 5 to 3
+move 13 from 6 to 1
+move 19 from 5 to 3
+move 12 from 3 to 9
+move 4 from 8 to 2
+move 2 from 6 to 7
+move 5 from 8 to 7
+move 9 from 3 to 9
+move 1 from 5 to 9
+move 2 from 7 to 6
+move 3 from 2 to 3
+move 15 from 9 to 3
+move 13 from 3 to 5
+move 1 from 6 to 2
+move 5 from 5 to 8
+move 1 from 2 to 5
+move 1 from 7 to 6
+move 6 from 9 to 6
+move 6 from 6 to 8
+move 4 from 7 to 1
+move 2 from 3 to 6
+move 11 from 1 to 9
+move 1 from 2 to 3
+move 4 from 5 to 6
+move 1 from 1 to 6
+move 10 from 9 to 2
+move 8 from 2 to 3
+move 3 from 1 to 2
+move 8 from 3 to 1
+move 5 from 5 to 4
+move 1 from 9 to 8
+move 2 from 3 to 7
+move 2 from 4 to 5
+move 6 from 1 to 6
+move 9 from 8 to 1
+move 16 from 1 to 9
+move 2 from 7 to 3
+move 3 from 3 to 8
+move 6 from 9 to 6
+move 1 from 5 to 4
+move 1 from 3 to 8
+move 5 from 2 to 1
+move 5 from 1 to 9
+move 2 from 4 to 9
+move 4 from 8 to 6
+move 1 from 8 to 7
+move 4 from 8 to 5
+move 2 from 8 to 2
+move 17 from 9 to 5
+move 11 from 5 to 7
+move 1 from 2 to 5
+move 1 from 2 to 5
+move 1 from 9 to 1
+move 1 from 1 to 6
+move 5 from 7 to 6
+move 20 from 6 to 7
+move 4 from 6 to 4
+move 15 from 7 to 8
+move 2 from 3 to 7
+move 1 from 6 to 5
+move 10 from 8 to 4
+move 1 from 3 to 6
+move 4 from 6 to 4
+move 13 from 7 to 8
+move 1 from 7 to 5
+move 1 from 6 to 3
+move 1 from 6 to 3
+move 1 from 6 to 9
+move 9 from 4 to 1
+move 3 from 8 to 2
+move 14 from 5 to 6
+move 2 from 2 to 8
+move 1 from 3 to 9
+move 14 from 6 to 2
+move 1 from 3 to 9
+move 1 from 9 to 3
+move 15 from 2 to 1
+move 1 from 3 to 9
+move 4 from 4 to 9
+move 10 from 8 to 5
+move 1 from 9 to 5
+move 1 from 1 to 5
+move 4 from 8 to 7
+move 3 from 9 to 3
+move 1 from 8 to 5
+move 1 from 4 to 7
+move 2 from 8 to 7
+move 6 from 5 to 6
+move 4 from 1 to 2
+move 1 from 2 to 5
+move 2 from 2 to 8
+move 2 from 8 to 1
+move 3 from 7 to 2
+move 3 from 4 to 9
+move 18 from 1 to 8
+move 1 from 7 to 3
+move 3 from 9 to 6
+move 1 from 1 to 5
+move 5 from 6 to 4
+move 2 from 1 to 9
+move 8 from 4 to 5
+move 4 from 3 to 2
+move 16 from 5 to 4
+move 8 from 8 to 6
+move 2 from 2 to 6
+move 1 from 7 to 6
+move 7 from 8 to 1
+move 1 from 2 to 3
+move 2 from 8 to 3
+move 4 from 4 to 9
+move 4 from 1 to 2
+move 1 from 7 to 2
+move 1 from 5 to 4
+move 1 from 3 to 7
+move 3 from 4 to 5
+move 1 from 9 to 6
+move 9 from 2 to 5
+move 2 from 3 to 6
+move 3 from 5 to 8
+move 3 from 1 to 7
+move 4 from 5 to 8
+move 1 from 4 to 3
+move 5 from 9 to 5
+move 5 from 5 to 8
+move 1 from 3 to 4
+move 4 from 5 to 1
+move 2 from 5 to 4
+move 13 from 6 to 2
+move 12 from 2 to 9
+move 3 from 9 to 2
+move 4 from 1 to 6
+move 8 from 6 to 2
+move 1 from 4 to 9
+move 3 from 7 to 9
+move 2 from 9 to 8
+move 1 from 7 to 2
+move 9 from 9 to 5
+move 2 from 8 to 6
+move 4 from 2 to 3
+move 1 from 7 to 2
+move 1 from 6 to 4
+move 4 from 3 to 9
+move 9 from 5 to 8
+move 10 from 4 to 2
+move 1 from 4 to 7
+move 1 from 6 to 2
+move 1 from 6 to 7
+move 13 from 2 to 6
+move 1 from 2 to 5
+move 6 from 6 to 5
+move 7 from 5 to 8
+move 1 from 4 to 5
+move 27 from 8 to 5
+move 3 from 6 to 3
+move 2 from 8 to 6
+move 8 from 9 to 5
+move 1 from 7 to 9
+move 1 from 6 to 2
+move 4 from 5 to 9
+move 2 from 3 to 4
+move 9 from 2 to 5
+move 1 from 4 to 1
+move 1 from 4 to 2
+move 1 from 2 to 4
+move 1 from 3 to 7
+move 1 from 1 to 3
+move 1 from 3 to 9
+move 6 from 9 to 4
+move 1 from 7 to 5
+move 13 from 5 to 2
+move 1 from 9 to 5
+move 1 from 7 to 2
+move 5 from 2 to 7
+move 8 from 5 to 7
+move 6 from 4 to 2
+move 1 from 4 to 5
+move 3 from 2 to 4
+move 4 from 2 to 7
+move 2 from 4 to 3
+move 13 from 7 to 3
+move 5 from 2 to 3
+move 4 from 7 to 8
+move 11 from 3 to 8
+move 11 from 5 to 9
+move 4 from 6 to 9
+move 1 from 6 to 5
+move 1 from 4 to 2
+move 1 from 3 to 6
+move 3 from 2 to 6
+move 3 from 6 to 2
+move 1 from 6 to 1
+move 1 from 3 to 8
+move 3 from 3 to 6
+move 2 from 2 to 7
+move 4 from 3 to 9
+move 16 from 9 to 2
+move 1 from 7 to 8
+move 2 from 2 to 8
+move 9 from 2 to 3
+move 6 from 2 to 7
+move 1 from 6 to 3
+move 2 from 9 to 2
+move 1 from 9 to 7
+move 2 from 6 to 3
+move 4 from 3 to 9
+move 2 from 2 to 7
+move 1 from 2 to 5
+move 14 from 5 to 6
+move 14 from 6 to 3
+move 4 from 9 to 8
+move 5 from 8 to 4
+move 1 from 1 to 5
+move 4 from 8 to 1
+move 1 from 5 to 9
+move 8 from 7 to 2
+move 18 from 3 to 7
+move 1 from 1 to 5
+move 1 from 1 to 9
+move 1 from 4 to 5
+move 1 from 8 to 5
+move 8 from 2 to 9
+move 3 from 5 to 8
+move 7 from 7 to 1
+move 3 from 4 to 7
+move 1 from 3 to 6
+move 7 from 8 to 3
+move 2 from 9 to 3
+move 3 from 8 to 9
+move 9 from 1 to 7
+move 9 from 3 to 4
+move 2 from 3 to 4
+move 12 from 7 to 4
+move 1 from 3 to 8
+move 1 from 8 to 7
+move 8 from 4 to 7
+move 11 from 4 to 9
+move 5 from 4 to 8
+move 19 from 7 to 9
+move 1 from 6 to 2
+move 2 from 7 to 4
+move 2 from 8 to 3
+move 1 from 7 to 8
+move 1 from 3 to 2
+move 3 from 8 to 4
+move 1 from 8 to 9
+move 1 from 3 to 2
+move 36 from 9 to 1
+move 5 from 9 to 6
+move 5 from 4 to 2
+move 24 from 1 to 3
+move 5 from 6 to 7
+move 1 from 1 to 4
+move 14 from 3 to 4
+move 4 from 7 to 3
+move 1 from 8 to 5
+move 5 from 2 to 9
+move 1 from 1 to 6
+move 5 from 9 to 1
+move 3 from 2 to 3
+move 1 from 5 to 3
+move 11 from 4 to 2
+move 1 from 7 to 1
+move 6 from 1 to 9
+move 3 from 4 to 2
+move 1 from 6 to 7
+move 10 from 1 to 7
+move 3 from 2 to 1
+move 3 from 3 to 2
+move 2 from 1 to 7
+move 1 from 4 to 8
+move 13 from 3 to 2
+move 1 from 8 to 3
+move 2 from 7 to 5
+move 2 from 3 to 7
+move 2 from 5 to 2
+move 1 from 1 to 7
+move 28 from 2 to 6
+move 1 from 2 to 3
+move 2 from 8 to 2
+move 6 from 9 to 7
+move 1 from 3 to 8
+move 1 from 9 to 8
+move 3 from 6 to 2
+move 14 from 7 to 9
+move 3 from 2 to 1
+move 2 from 2 to 9
+move 2 from 1 to 9
+move 1 from 9 to 1
+move 7 from 6 to 9
+move 2 from 1 to 4
+move 2 from 4 to 6
+move 4 from 8 to 7
+move 1 from 7 to 6
+move 1 from 8 to 1
+move 1 from 3 to 6
+move 1 from 1 to 5
+move 14 from 9 to 8
+move 1 from 5 to 9
+move 5 from 7 to 3
+move 16 from 6 to 3
+move 2 from 7 to 4
+move 8 from 9 to 5
+move 6 from 6 to 1
+move 8 from 5 to 9
+move 2 from 7 to 4
+move 11 from 9 to 1
+move 4 from 4 to 1
+move 14 from 8 to 3
+move 2 from 1 to 7
+move 20 from 3 to 6
+move 5 from 3 to 1
+move 1 from 3 to 5
+move 2 from 7 to 4
+move 20 from 6 to 7
+move 18 from 7 to 6
+move 17 from 6 to 9
+move 1 from 5 to 3
+move 6 from 3 to 2
+move 3 from 3 to 1
+move 1 from 6 to 2
+move 2 from 7 to 8
+move 4 from 1 to 5
+move 2 from 4 to 9
+move 1 from 3 to 2
+move 1 from 8 to 6
+move 18 from 1 to 4
+move 1 from 2 to 7
+move 1 from 6 to 2
+move 3 from 4 to 3
+move 1 from 8 to 1
+move 4 from 1 to 6
+move 7 from 2 to 1
+move 1 from 5 to 7
+move 1 from 4 to 1
+move 2 from 6 to 3
+move 3 from 5 to 9
+move 9 from 9 to 8
+move 10 from 9 to 3
+move 9 from 3 to 5