day 9 part 2 - simply run with input '2'

vaskoz · Dec 10, 2019 · b5d9cd7 · b5d9cd7
1 parent f870c58
commit b5d9cd7
Show file tree

Hide file tree

Showing 3 changed files with 327 additions and 0 deletions.
diff --git a/day9p2/input.txt b/day9p2/input.txt
@@ -0,0 +1 @@
+1102,34463338,34463338,63,1007,63,34463338,63,1005,63,53,1101,0,3,1000,109,988,209,12,9,1000,209,6,209,3,203,0,1008,1000,1,63,1005,63,65,1008,1000,2,63,1005,63,904,1008,1000,0,63,1005,63,58,4,25,104,0,99,4,0,104,0,99,4,17,104,0,99,0,0,1101,29,0,1010,1102,1,1,1021,1101,0,36,1002,1101,573,0,1026,1101,0,33,1012,1102,1,25,1004,1102,1,38,1000,1102,31,1,1003,1102,23,1,1006,1102,777,1,1028,1102,20,1,1011,1101,0,566,1027,1101,0,27,1009,1101,26,0,1005,1101,0,0,1020,1102,1,37,1014,1101,32,0,1001,1101,0,24,1007,1101,0,35,1018,1101,30,0,1017,1101,0,22,1008,1102,460,1,1023,1101,0,768,1029,1102,1,487,1024,1102,1,34,1013,1102,1,28,1015,1101,0,39,1019,1101,478,0,1025,1101,0,463,1022,1101,21,0,1016,109,9,1208,0,30,63,1005,63,201,1001,64,1,64,1105,1,203,4,187,1002,64,2,64,109,3,1201,-8,0,63,1008,63,24,63,1005,63,227,1001,64,1,64,1106,0,229,4,209,1002,64,2,64,109,-1,2108,32,-8,63,1005,63,245,1106,0,251,4,235,1001,64,1,64,1002,64,2,64,109,-11,2101,0,2,63,1008,63,35,63,1005,63,275,1001,64,1,64,1105,1,277,4,257,1002,64,2,64,109,3,2101,0,-1,63,1008,63,36,63,1005,63,303,4,283,1001,64,1,64,1106,0,303,1002,64,2,64,109,16,21108,40,40,-6,1005,1013,325,4,309,1001,64,1,64,1106,0,325,1002,64,2,64,109,-4,21102,41,1,-4,1008,1011,41,63,1005,63,351,4,331,1001,64,1,64,1105,1,351,1002,64,2,64,109,-15,2102,1,4,63,1008,63,24,63,1005,63,375,1001,64,1,64,1106,0,377,4,357,1002,64,2,64,109,6,1201,-2,0,63,1008,63,25,63,1005,63,403,4,383,1001,64,1,64,1106,0,403,1002,64,2,64,109,8,2102,1,-6,63,1008,63,22,63,1005,63,425,4,409,1106,0,429,1001,64,1,64,1002,64,2,64,109,-1,2108,27,-4,63,1005,63,447,4,435,1106,0,451,1001,64,1,64,1002,64,2,64,109,8,2105,1,2,1105,1,469,4,457,1001,64,1,64,1002,64,2,64,109,5,2105,1,-2,4,475,1001,64,1,64,1106,0,487,1002,64,2,64,109,-33,1202,7,1,63,1008,63,37,63,1005,63,507,1105,1,513,4,493,1001,64,1,64,1002,64,2,64,109,2,2107,25,10,63,1005,63,535,4,519,1001,64,1,64,1106,0,535,1002,64,2,64,109,30,21107,42,41,-9,1005,1016,551,1106,0,557,4,541,1001,64,1,64,1002,64,2,64,109,2,2106,0,0,1001,64,1,64,1105,1,575,4,563,1002,64,2,64,109,-19,1202,-7,1,63,1008,63,32,63,1005,63,601,4,581,1001,64,1,64,1105,1,601,1002,64,2,64,109,-2,1207,-1,27,63,1005,63,619,4,607,1106,0,623,1001,64,1,64,1002,64,2,64,109,2,21101,43,0,6,1008,1014,45,63,1005,63,647,1001,64,1,64,1106,0,649,4,629,1002,64,2,64,109,17,1205,-4,663,4,655,1106,0,667,1001,64,1,64,1002,64,2,64,109,4,1205,-9,683,1001,64,1,64,1106,0,685,4,673,1002,64,2,64,109,-17,21101,44,0,-2,1008,1010,44,63,1005,63,711,4,691,1001,64,1,64,1105,1,711,1002,64,2,64,109,1,21102,45,1,3,1008,1016,42,63,1005,63,735,1001,64,1,64,1105,1,737,4,717,1002,64,2,64,109,-9,1207,1,25,63,1005,63,753,1105,1,759,4,743,1001,64,1,64,1002,64,2,64,109,23,2106,0,1,4,765,1001,64,1,64,1106,0,777,1002,64,2,64,109,-3,1206,-3,789,1105,1,795,4,783,1001,64,1,64,1002,64,2,64,109,-13,2107,25,-4,63,1005,63,815,1001,64,1,64,1105,1,817,4,801,1002,64,2,64,109,-9,21108,46,44,10,1005,1012,833,1105,1,839,4,823,1001,64,1,64,1002,64,2,64,109,-4,1208,10,22,63,1005,63,857,4,845,1105,1,861,1001,64,1,64,1002,64,2,64,109,28,1206,-6,879,4,867,1001,64,1,64,1105,1,879,1002,64,2,64,109,-4,21107,47,48,-3,1005,1019,897,4,885,1105,1,901,1001,64,1,64,4,64,99,21102,27,1,1,21101,915,0,0,1106,0,922,21201,1,14615,1,204,1,99,109,3,1207,-2,3,63,1005,63,964,21201,-2,-1,1,21102,1,942,0,1106,0,922,22101,0,1,-1,21201,-2,-3,1,21101,957,0,0,1105,1,922,22201,1,-1,-2,1106,0,968,22101,0,-2,-2,109,-3,2105,1,0
diff --git a/day9p2/main.go b/day9p2/main.go
@@ -0,0 +1,294 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"io/ioutil"
+	"os"
+	"strconv"
+	"strings"
+)
+
+// nolint
+var (
+	in  io.Reader = os.Stdin
+	out io.Writer = os.Stdout
+)
+
+func main() {
+	f, _ := os.Open("input.txt")
+	defer f.Close()
+	data, _ := ioutil.ReadAll(f)
+	prog := string(data)
+	prog = strings.TrimSpace(prog)
+
+	strs := strings.Split(prog, ",")
+	opcodes := make([]int, 1000*len(strs))
+
+	for i, s := range strs {
+		res, _ := strconv.Atoi(s)
+		opcodes[i] = res
+	}
+
+	relativeBase := 0
+
+	for pc := 0; opcodes[pc] != 99; {
+		switch code := opcodes[pc]; code {
+		case 1:
+			val1 := opcodes[opcodes[pc+1]]
+			val2 := opcodes[opcodes[pc+2]]
+			outPos := opcodes[pc+3]
+			opcodes[outPos] = val1 + val2
+			pc += 4
+		case 2:
+			val1 := opcodes[opcodes[pc+1]]
+			val2 := opcodes[opcodes[pc+2]]
+			outPos := opcodes[pc+3]
+			opcodes[outPos] = val1 * val2
+			pc += 4
+		case 3:
+			var userInput int
+
+			fmt.Fscanf(in, "%d", &userInput)
+
+			writeTo := opcodes[pc+1]
+			opcodes[writeTo] = userInput
+			pc += 2
+		case 4:
+			readFrom := opcodes[pc+1]
+			fmt.Fprintln(out, opcodes[readFrom])
+
+			pc += 2
+		case 5: // jump-if-true
+			if opcodes[opcodes[pc+1]] != 0 {
+				pc = opcodes[opcodes[pc+2]]
+			} else {
+				pc += 3
+			}
+		case 6: // jump-if-false
+			if opcodes[opcodes[pc+1]] == 0 {
+				pc = opcodes[opcodes[pc+2]]
+			} else {
+				pc += 3
+			}
+		case 7: // less-than
+			if opcodes[opcodes[pc+1]] < opcodes[opcodes[pc+2]] {
+				opcodes[opcodes[pc+3]] = 1
+			} else {
+				opcodes[opcodes[pc+3]] = 0
+			}
+
+			pc += 4
+		case 8: // equals
+			if opcodes[opcodes[pc+1]] == opcodes[opcodes[pc+2]] {
+				opcodes[opcodes[pc+3]] = 1
+			} else {
+				opcodes[opcodes[pc+3]] = 0
+			}
+
+			pc += 4
+		case 9: // update relative base
+			relativeBase += opcodes[opcodes[pc+1]]
+
+			pc += 2
+		default: // deal with parameter modes
+			var val1, val2 int
+
+			op := code % 100
+			code /= 100
+
+			switch op {
+			case 1, 2:
+				if code%10 == 2 {
+					val1 = opcodes[relativeBase+opcodes[pc+1]]
+				} else if code%10 == 1 {
+					val1 = opcodes[pc+1]
+				} else {
+					val1 = opcodes[opcodes[pc+1]]
+				}
+
+				code /= 10
+				if code%10 == 2 {
+					val2 = opcodes[relativeBase+opcodes[pc+2]]
+				} else if code%10 == 1 {
+					val2 = opcodes[pc+2]
+				} else {
+					val2 = opcodes[opcodes[pc+2]]
+				}
+				code /= 10
+
+				var outPos int
+
+				if code%10 == 2 {
+					outPos = relativeBase + opcodes[pc+3]
+				} else {
+					outPos = opcodes[pc+3]
+				}
+
+				switch op {
+				case 1:
+					opcodes[outPos] = val1 + val2
+				case 2:
+					opcodes[outPos] = val1 * val2
+				}
+
+				pc += 4
+			case 3:
+				var userInput, writeTo int
+
+				fmt.Fscanf(in, "%d", &userInput)
+
+				if code == 1 {
+					writeTo = opcodes[pc+1]
+				} else if code == 2 {
+					writeTo = relativeBase + opcodes[pc+1]
+				}
+				opcodes[writeTo] = userInput
+				pc += 2
+
+			case 4:
+				if code == 1 {
+					fmt.Fprintln(out, opcodes[pc+1])
+				} else if code == 2 {
+					fmt.Fprintln(out, opcodes[relativeBase+opcodes[pc+1]])
+				}
+
+				pc += 2
+			case 5: // jump-if-true
+				var v1, v2 int
+				if code%10 == 2 {
+					v1 = opcodes[relativeBase+opcodes[pc+1]]
+				} else if code%10 == 1 {
+					v1 = opcodes[pc+1]
+				} else {
+					v1 = opcodes[opcodes[pc+1]]
+				}
+
+				code /= 10
+
+				if code%10 == 2 {
+					v2 = opcodes[relativeBase+opcodes[pc+2]]
+				} else if code%10 == 1 {
+					v2 = opcodes[pc+2]
+				} else {
+					v2 = opcodes[opcodes[pc+2]]
+				}
+
+				if v1 != 0 {
+					pc = v2
+				} else {
+					pc += 3
+				}
+			case 6: // jump-if-false
+				var v1, v2 int
+				if code%10 == 2 {
+					v1 = opcodes[relativeBase+opcodes[pc+1]]
+				} else if code%10 == 1 {
+					v1 = opcodes[pc+1]
+				} else {
+					v1 = opcodes[opcodes[pc+1]]
+				}
+
+				code /= 10
+				if code%10 == 2 {
+					v2 = opcodes[relativeBase+opcodes[pc+2]]
+				} else if code%10 == 1 {
+					v2 = opcodes[pc+2]
+				} else {
+					v2 = opcodes[opcodes[pc+2]]
+				}
+
+				if v1 == 0 {
+					pc = v2
+				} else {
+					pc += 3
+				}
+			case 7: // less-than
+				var v1, v2 int
+
+				if code%10 == 2 {
+					v1 = opcodes[relativeBase+opcodes[pc+1]]
+				} else if code%10 == 1 {
+					v1 = opcodes[pc+1]
+				} else {
+					v1 = opcodes[opcodes[pc+1]]
+				}
+
+				code /= 10
+
+				if code%10 == 2 {
+					v2 = opcodes[relativeBase+opcodes[pc+2]]
+				} else if code%10 == 1 {
+					v2 = opcodes[pc+2]
+				} else {
+					v2 = opcodes[opcodes[pc+2]]
+				}
+
+				code /= 10
+
+				var outPos int
+
+				if code%10 == 2 {
+					outPos = relativeBase + opcodes[pc+3]
+				} else {
+					outPos = opcodes[pc+3]
+				}
+
+				if v1 < v2 {
+					opcodes[outPos] = 1
+				} else {
+					opcodes[outPos] = 0
+				}
+
+				pc += 4
+			case 8: // equals
+				var v1, v2 int
+
+				if code%10 == 2 {
+					v1 = opcodes[relativeBase+opcodes[pc+1]]
+				} else if code%10 == 1 {
+					v1 = opcodes[pc+1]
+				} else {
+					v1 = opcodes[opcodes[pc+1]]
+				}
+
+				code /= 10
+
+				if code%10 == 2 {
+					v2 = opcodes[relativeBase+opcodes[pc+2]]
+				} else if code%10 == 1 {
+					v2 = opcodes[pc+2]
+				} else {
+					v2 = opcodes[opcodes[pc+2]]
+				}
+				code /= 10
+
+				var outPos int
+
+				if code%10 == 2 {
+					outPos = relativeBase + opcodes[pc+3]
+				} else {
+					outPos = opcodes[pc+3]
+				}
+
+				if v1 == v2 {
+					opcodes[outPos] = 1
+				} else {
+					opcodes[outPos] = 0
+				}
+
+				pc += 4
+			case 9:
+				if code%10 == 2 {
+					relativeBase += opcodes[relativeBase+opcodes[pc+1]]
+				} else if code%10 == 1 {
+					relativeBase += opcodes[pc+1]
+				} else {
+					relativeBase += opcodes[opcodes[pc+1]]
+				}
+
+				pc += 2
+			}
+		}
+	}
+}
diff --git a/day9p2/main_test.go b/day9p2/main_test.go
@@ -0,0 +1,32 @@
+package main
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+)
+
+func TestMain(t *testing.T) {
+	in = strings.NewReader("2\n")
+	buff := new(bytes.Buffer)
+	out = buff
+
+	main()
+
+	result := strings.TrimSpace(buff.String())
+	expected := `35920`
+
+	if strings.TrimSpace(expected) != result {
+		t.Errorf("Expected %v, got %v", expected, result)
+	}
+}
+
+func BenchmarkMain(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		in = strings.NewReader("2\n")
+		buff := new(bytes.Buffer)
+		out = buff
+
+		main()
+	}
+}