Implementations for day 14

challenge/day14/a.go

@@ -0,0 +1,95 @@
+package day14
+
+import (
+	"fmt"
+	"math"
+	"strings"
+
+	"github.com/nlowe/aoc2019/util"
+
+	"github.com/nlowe/aoc2019/challenge"
+	"github.com/spf13/cobra"
+)
+
+var A = &cobra.Command{
+	Use:   "14a",
+	Short: "Day 14, Problem A",
+	Run: func(_ *cobra.Command, _ []string) {
+		fmt.Printf("Answer: %d\n", a(challenge.FromFile()))
+	},
+}
+
+const (
+	componentOre  = "ORE"
+	componentFuel = "FUEL"
+)
+
+type component struct {
+	name  string
+	count int
+}
+
+type equation struct {
+	inputs []component
+	output component
+}
+
+func parseEquation(eqn string) equation {
+	parts := strings.Split(eqn, "=>")
+	inputs := strings.Split(parts[0], ",")
+
+	result := equation{}
+	for _, input := range inputs {
+		parts := strings.Split(strings.TrimSpace(input), " ")
+		result.inputs = append(result.inputs, component{
+			name:  strings.TrimSpace(parts[1]),
+			count: util.MustAtoI(strings.TrimSpace(parts[0])),
+		})
+	}
+
+	outputParts := strings.Split(strings.TrimSpace(parts[1]), " ")
+	result.output = component{
+		name:  strings.TrimSpace(outputParts[1]),
+		count: util.MustAtoI(strings.TrimSpace(outputParts[0])),
+	}
+
+	return result
+}
+
+func a(challenge *challenge.Input) int {
+	equations := map[string]equation{}
+	for line := range challenge.Lines() {
+		eqn := parseEquation(line)
+		equations[eqn.output.name] = eqn
+	}
+
+	return cost(componentFuel, 1, equations, map[string]int{})
+}
+
+func cost(name string, n int, equations map[string]equation, leftovers map[string]int) int {
+	if name == componentOre {
+		return n
+	}
+
+	if leftovers[name] >= n {
+		leftovers[name] -= n
+		return 0
+	}
+
+	if leftovers[name] > 0 {
+		n -= leftovers[name]
+		leftovers[name] = 0
+	}
+
+	totalOre := 0
+	eqn := equations[name]
+
+	iterationCount := int(math.Ceil(float64(n) / float64(eqn.output.count)))
+
+	for _, c := range eqn.inputs {
+		totalOre += cost(c.name, c.count*iterationCount, equations, leftovers)
+	}
+
+	leftovers[name] += eqn.output.count*iterationCount - n
+	return totalOre
+}

challenge/day14/a_test.go

@@ -0,0 +1,75 @@
+package day14
+
+import (
+	"testing"
+
+	"github.com/nlowe/aoc2019/challenge"
+	"github.com/stretchr/testify/require"
+)
+
+func TestA(t *testing.T) {
+	tests := []struct {
+		eqns string
+		cost int
+	}{
+		{eqns: `10 ORE => 10 A
+1 ORE => 1 B
+7 A, 1 B => 1 C
+7 A, 1 C => 1 D
+7 A, 1 D => 1 E
+7 A, 1 E => 1 FUEL`, cost: 31},
+		{eqns: `9 ORE => 2 A
+8 ORE => 3 B
+7 ORE => 5 C
+3 A, 4 B => 1 AB
+5 B, 7 C => 1 BC
+4 C, 1 A => 1 CA
+2 AB, 3 BC, 4 CA => 1 FUEL`, cost: 165},
+		{eqns: `157 ORE => 5 NZVS
+165 ORE => 6 DCFZ
+44 XJWVT, 5 KHKGT, 1 QDVJ, 29 NZVS, 9 GPVTF, 48 HKGWZ => 1 FUEL
+12 HKGWZ, 1 GPVTF, 8 PSHF => 9 QDVJ
+179 ORE => 7 PSHF
+177 ORE => 5 HKGWZ
+7 DCFZ, 7 PSHF => 2 XJWVT
+165 ORE => 2 GPVTF
+3 DCFZ, 7 NZVS, 5 HKGWZ, 10 PSHF => 8 KHKGT`, cost: 13312},
+		{eqns: `2 VPVL, 7 FWMGM, 2 CXFTF, 11 MNCFX => 1 STKFG
+17 NVRVD, 3 JNWZP => 8 VPVL
+53 STKFG, 6 MNCFX, 46 VJHF, 81 HVMC, 68 CXFTF, 25 GNMV => 1 FUEL
+22 VJHF, 37 MNCFX => 5 FWMGM
+139 ORE => 4 NVRVD
+144 ORE => 7 JNWZP
+5 MNCFX, 7 RFSQX, 2 FWMGM, 2 VPVL, 19 CXFTF => 3 HVMC
+5 VJHF, 7 MNCFX, 9 VPVL, 37 CXFTF => 6 GNMV
+145 ORE => 6 MNCFX
+1 NVRVD => 8 CXFTF
+1 VJHF, 6 MNCFX => 4 RFSQX
+176 ORE => 6 VJHF`, cost: 180697},
+		{eqns: `171 ORE => 8 CNZTR
+7 ZLQW, 3 BMBT, 9 XCVML, 26 XMNCP, 1 WPTQ, 2 MZWV, 1 RJRHP => 4 PLWSL
+114 ORE => 4 BHXH
+14 VRPVC => 6 BMBT
+6 BHXH, 18 KTJDG, 12 WPTQ, 7 PLWSL, 31 FHTLT, 37 ZDVW => 1 FUEL
+6 WPTQ, 2 BMBT, 8 ZLQW, 18 KTJDG, 1 XMNCP, 6 MZWV, 1 RJRHP => 6 FHTLT
+15 XDBXC, 2 LTCX, 1 VRPVC => 6 ZLQW
+13 WPTQ, 10 LTCX, 3 RJRHP, 14 XMNCP, 2 MZWV, 1 ZLQW => 1 ZDVW
+5 BMBT => 4 WPTQ
+189 ORE => 9 KTJDG
+1 MZWV, 17 XDBXC, 3 XCVML => 2 XMNCP
+12 VRPVC, 27 CNZTR => 2 XDBXC
+15 KTJDG, 12 BHXH => 5 XCVML
+3 BHXH, 2 VRPVC => 7 MZWV
+121 ORE => 7 VRPVC
+7 XCVML => 6 RJRHP
+5 BHXH, 4 VRPVC => 5 LTCX`, cost: 2210736},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.eqns, func(t *testing.T) {
+			input := challenge.FromLiteral(tt.eqns)
+
+			require.Equal(t, tt.cost, a(input))
+		})
+	}
+}

challenge/day14/b.go

@@ -0,0 +1,38 @@
+package day14
+
+import (
+	"fmt"
+
+	"github.com/nlowe/aoc2019/challenge"
+	"github.com/spf13/cobra"
+)
+
+var B = &cobra.Command{
+	Use:   "14b",
+	Short: "Day 14, Problem B",
+	Run: func(_ *cobra.Command, _ []string) {
+		fmt.Printf("Answer: %d\n", b(challenge.FromFile()))
+	},
+}
+
+func b(challenge *challenge.Input) int {
+	equations := map[string]equation{}
+	for line := range challenge.Lines() {
+		eqn := parseEquation(line)
+		equations[eqn.output.name] = eqn
+	}
+
+	lo := 1000000000000 / cost(componentFuel, 1, equations, map[string]int{})
+	hi := 1000000000000
+
+	for lo < hi {
+		mid := (lo + hi + 1) / 2
+		if cost(componentFuel, mid, equations, map[string]int{}) <= 1000000000000 {
+			lo = mid
+		} else {
+			hi = mid - 1
+		}
+	}
+
+	return lo
+}

challenge/day14/b_test.go

@@ -0,0 +1,62 @@
+package day14
+
+import (
+	"testing"
+
+	"github.com/nlowe/aoc2019/challenge"
+	"github.com/stretchr/testify/require"
+)
+
+func TestB(t *testing.T) {
+	tests := []struct {
+		eqns string
+		fuel int
+	}{
+		{eqns: `157 ORE => 5 NZVS
+165 ORE => 6 DCFZ
+44 XJWVT, 5 KHKGT, 1 QDVJ, 29 NZVS, 9 GPVTF, 48 HKGWZ => 1 FUEL
+12 HKGWZ, 1 GPVTF, 8 PSHF => 9 QDVJ
+179 ORE => 7 PSHF
+177 ORE => 5 HKGWZ
+7 DCFZ, 7 PSHF => 2 XJWVT
+165 ORE => 2 GPVTF
+3 DCFZ, 7 NZVS, 5 HKGWZ, 10 PSHF => 8 KHKGT`, fuel: 82892753},
+		{eqns: `2 VPVL, 7 FWMGM, 2 CXFTF, 11 MNCFX => 1 STKFG
+17 NVRVD, 3 JNWZP => 8 VPVL
+53 STKFG, 6 MNCFX, 46 VJHF, 81 HVMC, 68 CXFTF, 25 GNMV => 1 FUEL
+22 VJHF, 37 MNCFX => 5 FWMGM
+139 ORE => 4 NVRVD
+144 ORE => 7 JNWZP
+5 MNCFX, 7 RFSQX, 2 FWMGM, 2 VPVL, 19 CXFTF => 3 HVMC
+5 VJHF, 7 MNCFX, 9 VPVL, 37 CXFTF => 6 GNMV
+145 ORE => 6 MNCFX
+1 NVRVD => 8 CXFTF
+1 VJHF, 6 MNCFX => 4 RFSQX
+176 ORE => 6 VJHF`, fuel: 5586022},
+		{eqns: `171 ORE => 8 CNZTR
+7 ZLQW, 3 BMBT, 9 XCVML, 26 XMNCP, 1 WPTQ, 2 MZWV, 1 RJRHP => 4 PLWSL
+114 ORE => 4 BHXH
+14 VRPVC => 6 BMBT
+6 BHXH, 18 KTJDG, 12 WPTQ, 7 PLWSL, 31 FHTLT, 37 ZDVW => 1 FUEL
+6 WPTQ, 2 BMBT, 8 ZLQW, 18 KTJDG, 1 XMNCP, 6 MZWV, 1 RJRHP => 6 FHTLT
+15 XDBXC, 2 LTCX, 1 VRPVC => 6 ZLQW
+13 WPTQ, 10 LTCX, 3 RJRHP, 14 XMNCP, 2 MZWV, 1 ZLQW => 1 ZDVW
+5 BMBT => 4 WPTQ
+189 ORE => 9 KTJDG
+1 MZWV, 17 XDBXC, 3 XCVML => 2 XMNCP
+12 VRPVC, 27 CNZTR => 2 XDBXC
+15 KTJDG, 12 BHXH => 5 XCVML
+3 BHXH, 2 VRPVC => 7 MZWV
+121 ORE => 7 VRPVC
+7 XCVML => 6 RJRHP
+5 BHXH, 4 VRPVC => 5 LTCX`, fuel: 460664},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.eqns, func(t *testing.T) {
+			input := challenge.FromLiteral(tt.eqns)
+
+			require.Equal(t, tt.fuel, b(input))
+		})
+	}
+}

challenge/day14/input.txt

@@ -0,0 +1,58 @@
+4 JWXL => 8 SNBF
+23 MPZQF, 10 TXVW, 8 JWXL => 6 DXLB
+1 SNZDR, 5 XMWHC, 1 NJSC => 7 MHSB
+2 TDHD, 11 TXVW => 4 RFNZ
+2 VRCD, 1 FGZG, 3 JWXL, 1 HQTL, 2 MPZQF, 1 GTPJ, 5 HQNMK, 10 CQZQ => 9 QMTZB
+3 SRDB, 2 ZMVLP => 3 DHFD
+1 DFQGF => 1 CVXJR
+193 ORE => 3 TRWXF
+23 MFJMS, 4 HJXJH => 1 WVDF
+5 TRWXF => 5 RXFJ
+4 GZQH => 7 SNZDR
+160 ORE => 4 PLPF
+1 PLPF => 5 NJSC
+2 QKPZ, 2 JBWFL => 7 HBSC
+15 DXLB, 1 TDHD, 9 RFNZ => 5 DBRPW
+7 PLPF, 4 GMZH => 7 PVNX
+3 JWXL, 1 XWDNT, 4 CQZQ => 2 TPBXV
+2 SNZDR => 9 WQWT
+1 WMCF => 2 XWDNT
+1 DFQGF, 8 FGZG => 5 LMHJQ
+168 ORE => 9 GMZH
+18 PVNX, 3 RXFJ => 4 JBWFL
+5 WQWT => 1 CQZQ
+6 QMTZB, 28 NVWM, 8 LMHJQ, 1 SNBF, 15 PLPF, 3 KMXPQ, 43 WVDF, 52 SVNS => 1 FUEL
+164 ORE => 9 RXRMQ
+2 MFJMS, 1 HJXJH, 7 WVDF => 7 NXWC
+8 QDGBV, 1 WMCF, 2 MHSB => 6 HQTL
+1 XMWHC => 8 MLSK
+2 GMZH, 1 RXRMQ => 2 GZQH
+4 MPZQF, 7 WVDF => 9 KHJMV
+4 ZMVLP, 19 MLSK, 1 GZQH => 8 MFJMS
+1 HQTL, 1 SXKQ => 2 PWBKR
+3 SXKQ, 16 TXVW, 4 SVNS => 5 PSRF
+4 MPZQF, 3 SVNS => 9 QDGBV
+7 NXWC => 8 FGZG
+7 TDHD, 1 WQWT, 1 HBSC => 9 TXVW
+14 JBWFL => 5 LMXB
+1 VRCD, 3 KHJMV => 3 RTBL
+16 DHFD, 2 LBNK => 9 SXKQ
+1 QDGBV, 1 NJSC => 6 JWXL
+4 KHJMV => 3 HQNMK
+5 GZQH => 6 LBNK
+12 KHJMV, 19 FGZG, 3 XWDNT => 4 VRCD
+5 DHFD, 3 MLSK => 8 QKPZ
+4 KHJMV, 1 CQDR, 3 DBRPW, 2 CQZQ, 1 TPBXV, 15 TXVW, 2 TKSLM => 5 NVWM
+2 KHJMV => 5 CQDR
+1 CVXJR => 8 SVNS
+35 RXFJ, 5 NJSC, 22 PVNX => 9 HJXJH
+5 LMXB => 3 DFQGF
+1 RXFJ => 2 SRDB
+20 TPBXV, 1 RTBL, 13 PWBKR, 6 RFNZ, 1 LMXB, 2 CVXJR, 3 PSRF, 25 MPZQF => 9 KMXPQ
+1 MHSB, 8 MPZQF => 3 TDHD
+6 DHFD, 3 LBNK => 7 WMCF
+1 SRDB => 7 ZMVLP
+3 RXFJ => 8 XMWHC
+1 MPZQF => 8 TKSLM
+9 JBWFL, 22 WQWT => 8 MPZQF
+12 HBSC, 15 TKSLM => 1 GTPJ

main.go

@@ -4,6 +4,8 @@ import (
 	"fmt"
 	"time"
 
+	"github.com/nlowe/aoc2019/challenge/day14"
+
 	"github.com/nlowe/aoc2019/challenge/day13"
 
 	"github.com/nlowe/aoc2019/challenge/day1"
@@ -68,6 +70,7 @@ func init() {
 		day11.A, day11.B,
 		day12.A, day12.B,
 		day13.A, day13.B,
+		day14.A, day14.B,
 	)
 
 	flags := rootCmd.PersistentFlags()