From: Grégory Burri <gregory.burri@matisa.ch>
Date: Mon, 9 Dec 2019 15:40:37 +0000 (+0100)
Subject: Day[09]
X-Git-Url: https://git.euphorik.ch/?a=commitdiff_plain;h=a8e684a8e938af0dca87b4955395f98a45e3afd4;p=advent_of_code_2019.git
Day[09]
---
diff --git a/src/common.rs b/src/common.rs
index e87a0a2..30155d0 100644
--- a/src/common.rs
+++ b/src/common.rs
@@ -1,6 +1,13 @@
use std::fs;
use std::path::Path;
+use std::str::FromStr;
-pub fn read_list_of_numbers<P: AsRef<Path>>(file: P, sep: &str) -> Vec<i32> {
- fs::read_to_string(file).unwrap().split(sep).map(|line| line.parse::<i32>().unwrap()).collect()
+pub fn read_list_of_numbers<P, T>(file: P, sep: &str) -> Vec<T>
+where
+ P: AsRef<Path>,
+ T: FromStr,
+ T::Err: std::fmt::Debug
+
+{
+ fs::read_to_string(file).unwrap().split(sep).map(|line| line.parse::<T>().unwrap()).collect()
}
\ No newline at end of file
diff --git a/src/day07.rs b/src/day07.rs
new file mode 100644
index 0000000..77d8c48
--- /dev/null
+++ b/src/day07.rs
@@ -0,0 +1 @@
+fn fun() { }
\ No newline at end of file
diff --git a/src/day08.rs b/src/day08.rs
new file mode 100644
index 0000000..e69de29
diff --git a/src/day09.rs b/src/day09.rs
new file mode 100644
index 0000000..ea234b2
--- /dev/null
+++ b/src/day09.rs
@@ -0,0 +1,152 @@
+#[derive(Copy, Clone, Debug)]
+enum Mode {
+ Position,
+ Immediate,
+ Relative
+}
+
+// 'true' -> immediate mode, 'false' -> position mode.
+fn read_op_and_modes(mut code: i64) -> (i64, [Mode; 3]) {
+ let mut modes: [Mode; 3] = [ Mode::Position, Mode::Position, Mode::Position ];
+
+ for i in (2..=4).rev() {
+ let power = 10i64.pow(i);
+ if code >= 2 * power {
+ modes[i as usize - 2] = Mode::Relative;
+ code -= 2 * power;
+ } else if code >= power {
+ modes[i as usize - 2] = Mode::Immediate;
+ code -= power;
+ }
+ }
+
+ (code, modes)
+}
+
+pub fn execute_op_code(code: &[i64], input: &[i64]) -> Vec<i64> {
+ let mut code = Vec::from(code);
+ let mut cursor = 0;
+ let mut input_cursor = 0;
+ let mut output = Vec::<i64>::new();
+ let mut relative_base = 0;
+
+ fn read(position: usize, code: &[i64], mode: Mode, relative_base: i64) -> i64 {
+ if position >= code.len() { return 0 }
+
+ match mode {
+ Mode::Position => if code[position] as usize >= code.len() { 0 } else { code[code[position] as usize] },
+ Mode::Immediate => code[position],
+ Mode::Relative => if (code[position] + relative_base) as usize >= code.len() { 0 } else { code[(code[position] + relative_base) as usize] }
+ }
+ };
+
+ fn write(position: usize, value: i64, code: &mut Vec<i64>, mode: Mode, relative_base: i64) {
+ let ref_position =
+ match mode {
+ Mode::Position => code[position] as usize,
+ Mode::Immediate => panic!("Can't write un immediate mode"),
+ Mode::Relative => (code[position] + relative_base) as usize
+ };
+
+ if ref_position >= code.len() {
+ code.resize(ref_position + 1, 0);
+ }
+
+ code[ref_position] = value;
+ }
+
+ fn jump_if(cond: bool, cursor: usize, code: &[i64], modes: [Mode; 3], relative_base: i64) -> usize {
+ let value = read(cursor + 1, &code, modes[0], relative_base);
+ if cond == (value != 0) {
+ read(cursor + 2, &code, modes[1], relative_base) as usize
+ } else {
+ cursor + 3
+ }
+ };
+
+ loop {
+ let (op, modes) = read_op_and_modes(code[cursor]);
+
+ match op {
+ // Sum.
+ 1 => {
+ write(cursor + 3, read(cursor + 1, &code, modes[0], relative_base) + read(cursor + 2, &code, modes[1], relative_base), &mut code, modes[2], relative_base);
+ cursor += 4;
+ },
+
+ // Multiply.
+ 2 => {
+ write(cursor + 3, read(cursor + 1, &code, modes[0], relative_base) * read(cursor + 2, &code, modes[1], relative_base), &mut code, modes[2], relative_base);
+ cursor += 4;
+ },
+
+ // Input.
+ 3 => {
+ write(cursor + 1, input[input_cursor], &mut code, modes[0], relative_base);
+ input_cursor += 1;
+ cursor += 2;
+ }
+
+ // Output.
+ 4 => {
+ output.push(read(cursor + 1, &code, modes[0], relative_base));
+ cursor += 2;
+ }
+
+ // Jump-if-true.
+ 5 => cursor = jump_if(true, cursor, &code, modes, relative_base),
+
+ // Jump-if-false.
+ 6 => cursor = jump_if(false, cursor, &code, modes, relative_base),
+
+ // Less than.
+ 7 => {
+ write(cursor + 3, if read(cursor + 1, &code, modes[0], relative_base) < read(cursor + 2, &code, modes[1], relative_base) { 1 } else { 0 }, &mut code, modes[2], relative_base);
+ cursor += 4;
+ },
+
+ // Equals.
+ 8 => {
+ write(cursor + 3, if read(cursor + 1, &code, modes[0], relative_base) == read(cursor + 2, &code, modes[1], relative_base) { 1 } else { 0 }, &mut code, modes[2], relative_base);
+ cursor += 4;
+ },
+
+ // Change relative base.
+ 9 => {
+ relative_base += read(cursor + 1, &code, modes[0], relative_base);
+ cursor += 2;
+ }
+
+ 99 => break,
+
+ _ => panic!("Unkown code: {}", code[cursor])
+ }
+ }
+ output
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+
+ #[test]
+ fn copy_of_itself() {
+ let c = [109,1,204,-1,1001,100,1,100,1008,100,16,101,1006,101,0,99];
+ let r = execute_op_code(&c, &Vec::new());
+ assert_eq!(r, c);
+ }
+
+ #[test]
+ fn output_big_number() {
+ let c = [1102,34_915_192,34_915_192,7,4,7,99,0];
+ let r = execute_op_code(&c, &Vec::new());
+ assert_eq!(r[0], 1_219_070_632_396_864);
+ }
+
+ #[test]
+ fn input_big_number() {
+ let c = [104,1_125_899_906_842_624,99];
+ let r = execute_op_code(&c, &Vec::new());
+ assert_eq!(r[0], 1_125_899_906_842_624);
+ }
+}
\ No newline at end of file
diff --git a/src/main.rs b/src/main.rs
index eaccb71..2c96367 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,6 +7,9 @@ mod day02;
mod day03;
mod day05;
mod day06;
+mod day07;
+mod day08;
+mod day09;
mod common;
fn day01() -> String {
@@ -45,6 +48,19 @@ fn day06() -> String {
format!("part1: {}, part2: {}", day06::total_direct_and_indirect_orbits(&orbits), day06::nb_orbital_transfers(&orbits, "SAN", "YOU"))
}
+fn day07() -> String {
+ format!("")
+}
+
+fn day08() -> String {
+ format!("")
+}
+
+fn day09() -> String {
+ let code = common::read_list_of_numbers("data/day09.input", ",");
+ format!("part1: {:?}, part2: {:?}", day09::execute_op_code(&code, &[1]), day09::execute_op_code(&code, &[2]))
+}
+
fn format_micros(t: u128) -> String {
if t < 10_000 {
format!("{} μs", t)
@@ -69,7 +85,10 @@ fn main() {
day03,
day04,
day05,
- day06
+ day06,
+ day07,
+ day08,
+ day09
);
let args: Vec<String> = env::args().skip(1).collect();