Year 2019 Day 2

README.md

@@ -194,6 +194,7 @@ pie
 | Day | Problem | Solution | Benchmark (μs) |
 | --- | --- | --- | --: |
 | 1 | [The Tyranny of the Rocket Equation](https://adventofcode.com/2019/day/1) | [Source](src/year2019/day01.rs) | 1 |
+| 2 | [1202 Program Alarm](https://adventofcode.com/2019/day/2) | [Source](src/year2019/day02.rs) | 4 |
 
 ## 2015
 

benches/benchmark.rs

@@ -121,6 +121,7 @@ mod year2020 {
 
 mod year2019 {
     benchmark!(year2019, day01);
+    benchmark!(year2019, day02);
 }
 
 mod year2015 {

input/year2019/day02.txt

@@ -0,0 +1 @@
+1,0,0,3,1,1,2,3,1,3,4,3,1,5,0,3,2,10,1,19,1,6,19,23,1,23,13,27,2,6,27,31,1,5,31,35,2,10,35,39,1,6,39,43,1,13,43,47,2,47,6,51,1,51,5,55,1,55,6,59,2,59,10,63,1,63,6,67,2,67,10,71,1,71,9,75,2,75,10,79,1,79,5,83,2,10,83,87,1,87,6,91,2,9,91,95,1,95,5,99,1,5,99,103,1,103,10,107,1,9,107,111,1,6,111,115,1,115,5,119,1,10,119,123,2,6,123,127,2,127,6,131,1,131,2,135,1,10,135,0,99,2,0,14,0

src/lib.rs

@@ -174,6 +174,7 @@ pub mod year2020 {
 /// # Rescue Santa from deep space with a solar system adventure.
 pub mod year2019 {
     pub mod day01;
+    pub mod day02;
 }
 
 /// # Help Santa by solving puzzles to fix the weather machine's snow function.

src/main.rs

@@ -161,6 +161,7 @@ fn all_solutions() -> Vec<Solution> {
         solution!(year2020, day25),
         // 2019
         solution!(year2019, day01),
+        solution!(year2019, day02),
         // 2015
         solution!(year2015, day01),
         solution!(year2015, day02),

src/year2019/day02.rs

@@ -0,0 +1,59 @@
+//! # 1202 Program Alarm
+//!
+//! To speed up part two we binary search in two dimensions, instead of a brute force check of
+//! all possible 10,000 combinations.
+//!
+//! Since the only instructions are non-branching adds and multiplies the output is
+/// monotonically increasing in both dimensions.
+use crate::util::parse::*;
+use std::cmp::Ordering::*;
+
+pub fn parse(input: &str) -> Vec<usize> {
+    input.iter_unsigned().collect()
+}
+
+pub fn part1(input: &[usize]) -> usize {
+    check(input, 12, 2)
+}
+
+pub fn part2(input: &[usize]) -> i32 {
+    fn search(input: &[usize], x1: i32, x2: i32, y1: i32, y2: i32) -> Option<i32> {
+        if x1 > x2 || y1 > y2 {
+            return None;
+        }
+
+        let x = (x1 + x2) / 2;
+        let y = (y1 + y2) / 2;
+        let result = check(input, x as usize, y as usize);
+
+        match result.cmp(&19690720) {
+            Equal => Some(100 * x + y),
+            Less => search(input, x + 1, x2, y1, y2).or_else(|| search(input, x1, x2, y + 1, y2)),
+            Greater => {
+                search(input, x1, x - 1, y1, y2).or_else(|| search(input, x1, x2, y1, y - 1))
+            }
+        }
+    }
+    search(input, 0, 99, 0, 99).unwrap()
+}
+
+fn check(input: &[usize], first: usize, second: usize) -> usize {
+    let code = &mut input.to_vec();
+    code[1] = first;
+    code[2] = second;
+
+    execute(code)
+}
+
+fn execute(code: &mut [usize]) -> usize {
+    let mut pc = 0;
+
+    loop {
+        match code[pc] {
+            1 => code[code[pc + 3]] = code[code[pc + 1]] + code[code[pc + 2]],
+            2 => code[code[pc + 3]] = code[code[pc + 1]] * code[code[pc + 2]],
+            _ => break code[0],
+        }
+        pc += 4;
+    }
+}

tests/test.rs

@@ -84,6 +84,7 @@ mod year2020 {
 
 mod year2019 {
     mod day01_test;
+    mod day02_test;
 }
 
 mod year2015 {

tests/year2019/day02_test.rs

@@ -0,0 +1,14 @@
+use aoc::year2019::day02::*;
+
+const EXAMPLE: &str = "1, 9, 10, 3, 2, 3, 11, 0, 99, 30, 40, 50, 0";
+
+#[test]
+fn part1_test() {
+    let input = parse(EXAMPLE);
+    assert_eq!(part1(&input), 100);
+}
+
+#[test]
+fn part2_test() {
+    // Integration test not possible with example data.
+}