Completed day 7 part 1
This commit is contained in:
parent
c234561337
commit
74d77382dd
144
day7/Cargo.lock
generated
Normal file
144
day7/Cargo.lock
generated
Normal file
@ -0,0 +1,144 @@
|
|||||||
|
# This file is automatically @generated by Cargo.
|
||||||
|
# It is not intended for manual editing.
|
||||||
|
[[package]]
|
||||||
|
name = "day7"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"num_enum 0.4.2 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"permutohedron 0.2.4 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "derivative"
|
||||||
|
version = "1.0.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2 0.4.30 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"quote 0.6.13 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"syn 0.15.44 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "num_enum"
|
||||||
|
version = "0.4.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"derivative 1.0.3 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"num_enum_derive 0.4.2 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "num_enum_derive"
|
||||||
|
version = "0.4.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro-crate 0.1.4 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"proc-macro2 1.0.6 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"quote 1.0.2 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"syn 1.0.11 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "permutohedron"
|
||||||
|
version = "0.2.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro-crate"
|
||||||
|
version = "0.1.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"toml 0.5.5 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro2"
|
||||||
|
version = "0.4.30"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-xid 0.1.0 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro2"
|
||||||
|
version = "1.0.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-xid 0.2.0 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "quote"
|
||||||
|
version = "0.6.13"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2 0.4.30 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "quote"
|
||||||
|
version = "1.0.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2 1.0.6 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde"
|
||||||
|
version = "1.0.103"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "syn"
|
||||||
|
version = "0.15.44"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2 0.4.30 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"quote 0.6.13 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"unicode-xid 0.1.0 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "syn"
|
||||||
|
version = "1.0.11"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2 1.0.6 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"quote 1.0.2 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
"unicode-xid 0.2.0 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "toml"
|
||||||
|
version = "0.5.5"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
dependencies = [
|
||||||
|
"serde 1.0.103 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "unicode-xid"
|
||||||
|
version = "0.1.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "unicode-xid"
|
||||||
|
version = "0.2.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
|
||||||
|
[metadata]
|
||||||
|
"checksum derivative 1.0.3 (registry+https://github.com/rust-lang/crates.io-index)" = "942ca430eef7a3806595a6737bc388bf51adb888d3fc0dd1b50f1c170167ee3a"
|
||||||
|
"checksum num_enum 0.4.2 (registry+https://github.com/rust-lang/crates.io-index)" = "be601e38e20a6f3d01049d85801cb9b7a34a8da7a0da70df507bbde7735058c8"
|
||||||
|
"checksum num_enum_derive 0.4.2 (registry+https://github.com/rust-lang/crates.io-index)" = "b59f30f6a043f2606adbd0addbf1eef6f2e28e8c4968918b63b7ff97ac0db2a7"
|
||||||
|
"checksum permutohedron 0.2.4 (registry+https://github.com/rust-lang/crates.io-index)" = "b687ff7b5da449d39e418ad391e5e08da53ec334903ddbb921db208908fc372c"
|
||||||
|
"checksum proc-macro-crate 0.1.4 (registry+https://github.com/rust-lang/crates.io-index)" = "e10d4b51f154c8a7fb96fd6dad097cb74b863943ec010ac94b9fd1be8861fe1e"
|
||||||
|
"checksum proc-macro2 0.4.30 (registry+https://github.com/rust-lang/crates.io-index)" = "cf3d2011ab5c909338f7887f4fc896d35932e29146c12c8d01da6b22a80ba759"
|
||||||
|
"checksum proc-macro2 1.0.6 (registry+https://github.com/rust-lang/crates.io-index)" = "9c9e470a8dc4aeae2dee2f335e8f533e2d4b347e1434e5671afc49b054592f27"
|
||||||
|
"checksum quote 0.6.13 (registry+https://github.com/rust-lang/crates.io-index)" = "6ce23b6b870e8f94f81fb0a363d65d86675884b34a09043c81e5562f11c1f8e1"
|
||||||
|
"checksum quote 1.0.2 (registry+https://github.com/rust-lang/crates.io-index)" = "053a8c8bcc71fcce321828dc897a98ab9760bef03a4fc36693c231e5b3216cfe"
|
||||||
|
"checksum serde 1.0.103 (registry+https://github.com/rust-lang/crates.io-index)" = "1217f97ab8e8904b57dd22eb61cde455fa7446a9c1cf43966066da047c1f3702"
|
||||||
|
"checksum syn 0.15.44 (registry+https://github.com/rust-lang/crates.io-index)" = "9ca4b3b69a77cbe1ffc9e198781b7acb0c7365a883670e8f1c1bc66fba79a5c5"
|
||||||
|
"checksum syn 1.0.11 (registry+https://github.com/rust-lang/crates.io-index)" = "dff0acdb207ae2fe6d5976617f887eb1e35a2ba52c13c7234c790960cdad9238"
|
||||||
|
"checksum toml 0.5.5 (registry+https://github.com/rust-lang/crates.io-index)" = "01d1404644c8b12b16bfcffa4322403a91a451584daaaa7c28d3152e6cbc98cf"
|
||||||
|
"checksum unicode-xid 0.1.0 (registry+https://github.com/rust-lang/crates.io-index)" = "fc72304796d0818e357ead4e000d19c9c174ab23dc11093ac919054d20a6a7fc"
|
||||||
|
"checksum unicode-xid 0.2.0 (registry+https://github.com/rust-lang/crates.io-index)" = "826e7639553986605ec5979c7dd957c7895e93eabed50ab2ffa7f6128a75097c"
|
||||||
11
day7/Cargo.toml
Normal file
11
day7/Cargo.toml
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
[package]
|
||||||
|
name = "day7"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Tyler Hallada <tyler@hallada.net>"]
|
||||||
|
edition = "2018"
|
||||||
|
|
||||||
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
num_enum = "0.4.2"
|
||||||
|
permutohedron = "0.2.4"
|
||||||
1
day7/input/input.txt
Normal file
1
day7/input/input.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
3,8,1001,8,10,8,105,1,0,0,21,38,63,76,93,118,199,280,361,442,99999,3,9,101,3,9,9,102,3,9,9,101,4,9,9,4,9,99,3,9,1002,9,2,9,101,5,9,9,1002,9,5,9,101,5,9,9,1002,9,4,9,4,9,99,3,9,101,2,9,9,102,3,9,9,4,9,99,3,9,101,2,9,9,102,5,9,9,1001,9,5,9,4,9,99,3,9,102,4,9,9,1001,9,3,9,1002,9,5,9,101,2,9,9,1002,9,2,9,4,9,99,3,9,1002,9,2,9,4,9,3,9,1001,9,1,9,4,9,3,9,1001,9,1,9,4,9,3,9,1001,9,1,9,4,9,3,9,1001,9,2,9,4,9,3,9,1002,9,2,9,4,9,3,9,101,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,1001,9,1,9,4,9,3,9,101,2,9,9,4,9,99,3,9,102,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,1001,9,2,9,4,9,3,9,102,2,9,9,4,9,3,9,101,1,9,9,4,9,3,9,102,2,9,9,4,9,3,9,102,2,9,9,4,9,3,9,1001,9,1,9,4,9,3,9,102,2,9,9,4,9,3,9,1001,9,1,9,4,9,99,3,9,101,1,9,9,4,9,3,9,101,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,101,2,9,9,4,9,3,9,1001,9,2,9,4,9,3,9,1002,9,2,9,4,9,3,9,1002,9,2,9,4,9,3,9,102,2,9,9,4,9,3,9,1001,9,1,9,4,9,3,9,1002,9,2,9,4,9,99,3,9,1001,9,1,9,4,9,3,9,102,2,9,9,4,9,3,9,102,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,1001,9,2,9,4,9,3,9,102,2,9,9,4,9,3,9,101,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,101,1,9,9,4,9,3,9,1001,9,2,9,4,9,99,3,9,1002,9,2,9,4,9,3,9,102,2,9,9,4,9,3,9,101,2,9,9,4,9,3,9,101,1,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,1001,9,2,9,4,9,3,9,102,2,9,9,4,9,3,9,101,1,9,9,4,9,3,9,101,2,9,9,4,9,3,9,1002,9,2,9,4,9,99
|
||||||
1
day7/input/test1.txt
Normal file
1
day7/input/test1.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
3,15,3,16,1002,16,10,16,1,16,15,15,4,15,99,0,0
|
||||||
1
day7/input/test2.txt
Normal file
1
day7/input/test2.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
3,23,3,24,1002,24,10,24,1002,23,-1,23,101,5,23,23,1,24,23,23,4,23,99,0,0
|
||||||
1
day7/input/test3.txt
Normal file
1
day7/input/test3.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
3,31,3,32,1002,32,10,32,1001,31,-2,31,1007,31,0,33,1002,33,7,33,1,33,31,31,1,32,31,31,4,31,99,0,0,0
|
||||||
351
day7/src/intcode.rs
Normal file
351
day7/src/intcode.rs
Normal file
@ -0,0 +1,351 @@
|
|||||||
|
use std::collections::VecDeque;
|
||||||
|
use std::convert::TryFrom;
|
||||||
|
use std::error::Error;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::prelude::*;
|
||||||
|
use std::result;
|
||||||
|
use std::str::FromStr;
|
||||||
|
|
||||||
|
use num_enum::TryFromPrimitive;
|
||||||
|
|
||||||
|
type Result<T> = result::Result<T, Box<dyn Error>>;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
|
pub struct Intcode {
|
||||||
|
integers: Vec<i32>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq)]
|
||||||
|
pub struct Instruction {
|
||||||
|
opcode: Opcode,
|
||||||
|
parameter_modes: Vec<ParameterMode>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TryFrom<i32> for Instruction {
|
||||||
|
type Error = Box<dyn Error>;
|
||||||
|
|
||||||
|
fn try_from(integer: i32) -> Result<Self> {
|
||||||
|
let opcode: Opcode = Opcode::try_from((integer % 100) as u8)?;
|
||||||
|
let modes_integer = integer / 100;
|
||||||
|
let mut parameter_modes = vec![];
|
||||||
|
for parameter_index in 0..opcode.parameter_count() {
|
||||||
|
parameter_modes.push(match opcode.target_parameter_index() {
|
||||||
|
Some(target_parameter_index)
|
||||||
|
if target_parameter_index == parameter_index as usize =>
|
||||||
|
{
|
||||||
|
ParameterMode::Position
|
||||||
|
}
|
||||||
|
_ => ParameterMode::try_from(
|
||||||
|
(modes_integer % (10_i32.pow(parameter_index + 1))
|
||||||
|
/ 10_i32.pow(parameter_index)) as u8,
|
||||||
|
)?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
Ok(Instruction {
|
||||||
|
opcode,
|
||||||
|
parameter_modes,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, TryFromPrimitive)]
|
||||||
|
#[repr(u8)]
|
||||||
|
pub enum Opcode {
|
||||||
|
Add = 1,
|
||||||
|
Mult = 2,
|
||||||
|
Input = 3,
|
||||||
|
Output = 4,
|
||||||
|
JumpIfTrue = 5,
|
||||||
|
JumpIfFalse = 6,
|
||||||
|
LessThan = 7,
|
||||||
|
Equals = 8,
|
||||||
|
Halt = 99,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Opcode {
|
||||||
|
pub fn parameter_count(&self) -> u32 {
|
||||||
|
match self {
|
||||||
|
Opcode::Add => 3,
|
||||||
|
Opcode::Mult => 3,
|
||||||
|
Opcode::Input => 1,
|
||||||
|
Opcode::Output => 1,
|
||||||
|
Opcode::JumpIfTrue => 2,
|
||||||
|
Opcode::JumpIfFalse => 2,
|
||||||
|
Opcode::LessThan => 3,
|
||||||
|
Opcode::Equals => 3,
|
||||||
|
Opcode::Halt => 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn target_parameter_index(&self) -> Option<usize> {
|
||||||
|
match self {
|
||||||
|
Opcode::Add => Some(2),
|
||||||
|
Opcode::Mult => Some(2),
|
||||||
|
Opcode::Input => Some(0),
|
||||||
|
Opcode::Output => None,
|
||||||
|
Opcode::JumpIfTrue => None,
|
||||||
|
Opcode::JumpIfFalse => None,
|
||||||
|
Opcode::LessThan => Some(2),
|
||||||
|
Opcode::Equals => Some(2),
|
||||||
|
Opcode::Halt => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, TryFromPrimitive)]
|
||||||
|
#[repr(u8)]
|
||||||
|
pub enum ParameterMode {
|
||||||
|
Position = 0,
|
||||||
|
Immediate = 1,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FromStr for Intcode {
|
||||||
|
type Err = Box<dyn Error>;
|
||||||
|
|
||||||
|
fn from_str(s: &str) -> Result<Intcode> {
|
||||||
|
let intcode_string = s.trim().to_string();
|
||||||
|
|
||||||
|
Ok(Intcode {
|
||||||
|
integers: intcode_string
|
||||||
|
.split(',')
|
||||||
|
.map(|code| code.parse().unwrap())
|
||||||
|
.collect(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Intcode {
|
||||||
|
fn load_parameters(&self, pointer: usize, instruction: &Instruction) -> Vec<i32> {
|
||||||
|
(0..instruction.opcode.parameter_count() as usize)
|
||||||
|
.map(|parameter_index| {
|
||||||
|
let mut integer = self.integers[pointer + parameter_index + 1];
|
||||||
|
if let ParameterMode::Position = instruction.parameter_modes[parameter_index] {
|
||||||
|
match instruction.opcode.target_parameter_index() {
|
||||||
|
Some(target_parameter_index)
|
||||||
|
if target_parameter_index == parameter_index => {}
|
||||||
|
_ => {
|
||||||
|
integer = self.integers[integer as usize];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
integer
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn execute(&mut self, inputs: &[i32]) -> Result<Vec<i32>> {
|
||||||
|
let mut pointer = 0;
|
||||||
|
let mut input_index = 0;
|
||||||
|
let mut output = vec![];
|
||||||
|
|
||||||
|
loop {
|
||||||
|
let instruction = Instruction::try_from(self.integers[pointer])?;
|
||||||
|
let parameters = self.load_parameters(pointer, &instruction);
|
||||||
|
let mut jump_pointer: Option<usize> = None;
|
||||||
|
|
||||||
|
match instruction.opcode {
|
||||||
|
Opcode::Add => {
|
||||||
|
self.integers[parameters[2] as usize] = parameters[0] + parameters[1];
|
||||||
|
}
|
||||||
|
Opcode::Mult => {
|
||||||
|
self.integers[parameters[2] as usize] = parameters[0] * parameters[1];
|
||||||
|
}
|
||||||
|
Opcode::Input => {
|
||||||
|
self.integers[parameters[0] as usize] = inputs[input_index];
|
||||||
|
input_index += 1;
|
||||||
|
}
|
||||||
|
Opcode::Output => {
|
||||||
|
output.push(parameters[0]);
|
||||||
|
}
|
||||||
|
Opcode::JumpIfTrue => {
|
||||||
|
if parameters[0] != 0 {
|
||||||
|
jump_pointer = Some(parameters[1] as usize);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Opcode::JumpIfFalse => {
|
||||||
|
if parameters[0] == 0 {
|
||||||
|
jump_pointer = Some(parameters[1] as usize);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Opcode::LessThan => {
|
||||||
|
if parameters[0] < parameters[1] {
|
||||||
|
self.integers[parameters[2] as usize] = 1;
|
||||||
|
} else {
|
||||||
|
self.integers[parameters[2] as usize] = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Opcode::Equals => {
|
||||||
|
if parameters[0] == parameters[1] {
|
||||||
|
self.integers[parameters[2] as usize] = 1;
|
||||||
|
} else {
|
||||||
|
self.integers[parameters[2] as usize] = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Opcode::Halt => {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
match jump_pointer {
|
||||||
|
Some(jump_pointer) => pointer = jump_pointer,
|
||||||
|
None => pointer += 1 + instruction.opcode.parameter_count() as usize,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(output)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn read_intcode(filename: &str) -> Result<Intcode> {
|
||||||
|
let mut file = File::open(filename)?;
|
||||||
|
let mut intcode_string = String::new();
|
||||||
|
file.read_to_string(&mut intcode_string)?;
|
||||||
|
|
||||||
|
Ok(intcode_string.parse()?)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
const TEST_INPUT: &str = "input/test1.txt";
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn reads_intcode() {
|
||||||
|
assert_eq!(
|
||||||
|
read_intcode(TEST_INPUT).unwrap(),
|
||||||
|
Intcode {
|
||||||
|
integers: vec![3, 15, 3, 16, 1002, 16, 10, 16, 1, 16, 15, 15, 4, 15, 99, 0, 0]
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn converts_integer_to_instruction() {
|
||||||
|
assert_eq!(
|
||||||
|
Instruction::try_from(1002).unwrap(),
|
||||||
|
Instruction {
|
||||||
|
opcode: Opcode::Mult,
|
||||||
|
parameter_modes: vec![
|
||||||
|
ParameterMode::Position,
|
||||||
|
ParameterMode::Immediate,
|
||||||
|
ParameterMode::Position
|
||||||
|
],
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
Instruction::try_from(101).unwrap(),
|
||||||
|
Instruction {
|
||||||
|
opcode: Opcode::Add,
|
||||||
|
parameter_modes: vec![
|
||||||
|
ParameterMode::Immediate,
|
||||||
|
ParameterMode::Position,
|
||||||
|
ParameterMode::Position
|
||||||
|
],
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn executes_intcodes() {
|
||||||
|
let mut intcode = Intcode {
|
||||||
|
integers: vec![1, 0, 0, 0, 99],
|
||||||
|
};
|
||||||
|
intcode.execute(&[0]).unwrap();
|
||||||
|
assert_eq!(intcode.integers, vec![2, 0, 0, 0, 99]);
|
||||||
|
|
||||||
|
let mut intcode = Intcode {
|
||||||
|
integers: vec![2, 3, 0, 3, 99],
|
||||||
|
};
|
||||||
|
intcode.execute(&[0]).unwrap();
|
||||||
|
assert_eq!(intcode.integers, vec![2, 3, 0, 6, 99]);
|
||||||
|
|
||||||
|
let mut intcode = Intcode {
|
||||||
|
integers: vec![2, 4, 4, 5, 99, 0],
|
||||||
|
};
|
||||||
|
intcode.execute(&[0]).unwrap();
|
||||||
|
assert_eq!(intcode.integers, vec![2, 4, 4, 5, 99, 9801]);
|
||||||
|
|
||||||
|
let mut intcode = Intcode {
|
||||||
|
integers: vec![1, 1, 1, 4, 99, 5, 6, 0, 99],
|
||||||
|
};
|
||||||
|
intcode.execute(&[0]).unwrap();
|
||||||
|
assert_eq!(intcode.integers, vec![30, 1, 1, 4, 2, 5, 6, 0, 99]);
|
||||||
|
|
||||||
|
let mut intcode = Intcode {
|
||||||
|
integers: vec![1, 9, 10, 3, 2, 3, 11, 0, 99, 30, 40, 50],
|
||||||
|
};
|
||||||
|
intcode.execute(&[0]).unwrap();
|
||||||
|
assert_eq!(
|
||||||
|
intcode.integers,
|
||||||
|
vec![3500, 9, 10, 70, 2, 3, 11, 0, 99, 30, 40, 50]
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn less_and_equal_outputs() {
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![3, 9, 8, 9, 10, 9, 4, 9, 99, -1, 8],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[8]).unwrap(), vec![1]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[0]).unwrap(), vec![0]);
|
||||||
|
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![3, 9, 7, 9, 10, 9, 4, 9, 99, -1, 8],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[0]).unwrap(), vec![1]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[9]).unwrap(), vec![0]);
|
||||||
|
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![3, 3, 1108, -1, 8, 3, 4, 3, 99],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[8]).unwrap(), vec![1]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[0]).unwrap(), vec![0]);
|
||||||
|
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![3, 3, 1107, -1, 8, 3, 4, 3, 99],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[0]).unwrap(), vec![1]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[9]).unwrap(), vec![0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn jump_outputs() {
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![3, 12, 6, 12, 15, 1, 13, 14, 13, 4, 13, 99, -1, 0, 1, 9],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[0]).unwrap(), vec![0]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[1]).unwrap(), vec![1]);
|
||||||
|
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![3, 3, 1105, -1, 9, 1101, 0, 0, 12, 4, 12, 99, 1],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[0]).unwrap(), vec![0]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[1]).unwrap(), vec![1]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn larger_part2_intcode() {
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![
|
||||||
|
3, 21, 1008, 21, 8, 20, 1005, 20, 22, 107, 8, 21, 20, 1006, 20, 31, 1106, 0, 36,
|
||||||
|
98, 0, 0, 1002, 21, 125, 20, 4, 20, 1105, 1, 46, 104, 999, 1105, 1, 46, 1101, 1000,
|
||||||
|
1, 20, 4, 20, 1105, 1, 46, 98, 99,
|
||||||
|
],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[0]).unwrap(), vec![999]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[8]).unwrap(), vec![1000]);
|
||||||
|
assert_eq!(intcode.clone().execute(&[9]).unwrap(), vec![1001]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn multiple_input_intcode() {
|
||||||
|
let intcode = Intcode {
|
||||||
|
integers: vec![
|
||||||
|
3, 15, 3, 16, 1002, 16, 10, 16, 1, 16, 15, 15, 4, 15, 99, 0, 0,
|
||||||
|
],
|
||||||
|
};
|
||||||
|
assert_eq!(intcode.clone().execute(&[1, 1]).unwrap(), vec![11]);
|
||||||
|
}
|
||||||
|
}
|
||||||
136
day7/src/main.rs
Normal file
136
day7/src/main.rs
Normal file
@ -0,0 +1,136 @@
|
|||||||
|
use std::error::Error;
|
||||||
|
use std::result;
|
||||||
|
|
||||||
|
use permutohedron::Heap;
|
||||||
|
|
||||||
|
mod intcode;
|
||||||
|
|
||||||
|
use intcode::{read_intcode, Intcode};
|
||||||
|
|
||||||
|
const INPUT: &str = "input/input.txt";
|
||||||
|
|
||||||
|
type Result<T> = result::Result<T, Box<dyn Error>>;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
|
struct Amplifier {
|
||||||
|
intcode: Intcode,
|
||||||
|
phase_setting: i32,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Amplifier {
|
||||||
|
fn new(intcode: Intcode, phase_setting: i32) -> Amplifier {
|
||||||
|
Amplifier {
|
||||||
|
intcode: intcode,
|
||||||
|
phase_setting: phase_setting,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn execute(&self, input: i32) -> Result<i32> {
|
||||||
|
let mut intcode = self.intcode.clone();
|
||||||
|
let output = intcode.execute(&[self.phase_setting, input])?;
|
||||||
|
dbg!(&output);
|
||||||
|
Ok(output[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
|
struct AmplificationCircuit {
|
||||||
|
amplifiers: Vec<Amplifier>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AmplificationCircuit {
|
||||||
|
fn new(
|
||||||
|
intcode: Intcode,
|
||||||
|
amplifier_count: usize,
|
||||||
|
initial_phase_settings: &[i32],
|
||||||
|
) -> AmplificationCircuit {
|
||||||
|
AmplificationCircuit {
|
||||||
|
amplifiers: (0..amplifier_count)
|
||||||
|
.map(|index| Amplifier::new(intcode.clone(), initial_phase_settings[index]))
|
||||||
|
.collect(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_phase_settings(&mut self, phase_settings: &[i32; 5]) {
|
||||||
|
for (index, phase_setting) in phase_settings.iter().enumerate() {
|
||||||
|
self.amplifiers[index].phase_setting = *phase_setting;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn execute_circuit(&self, input_signal: i32) -> Result<i32> {
|
||||||
|
let mut input = input_signal;
|
||||||
|
for amplifier in self.amplifiers.iter() {
|
||||||
|
input = amplifier.execute(input)?;
|
||||||
|
}
|
||||||
|
Ok(input)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn find_max_output(&mut self, input_signal: i32) -> Result<i32> {
|
||||||
|
let mut phase_setting: [i32; 5] = [0, 1, 2, 3, 4];
|
||||||
|
let mut max_output = 0;
|
||||||
|
let heap = Heap::new(&mut phase_setting);
|
||||||
|
|
||||||
|
for permutation in heap {
|
||||||
|
self.set_phase_settings(&permutation);
|
||||||
|
|
||||||
|
let output = self.execute_circuit(input_signal)?;
|
||||||
|
if output > max_output {
|
||||||
|
max_output = output;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(max_output)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn solve_part1() -> Result<i32> {
|
||||||
|
let intcode = read_intcode(INPUT)?;
|
||||||
|
let mut circuit = AmplificationCircuit::new(intcode, 5, &[0, 0, 0, 0, 0]);
|
||||||
|
Ok(circuit.find_max_output(0)?)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn solve_part2() -> Result<i32> {
|
||||||
|
Ok(0)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() -> Result<()> {
|
||||||
|
println!("Part 1: {}", solve_part1()?);
|
||||||
|
println!("Part 2: {}", solve_part2()?);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
const TEST_INPUT1: &str = "input/test1.txt";
|
||||||
|
const TEST_INPUT2: &str = "input/test2.txt";
|
||||||
|
const TEST_INPUT3: &str = "input/test3.txt";
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn executes_amplifier_circuits() {
|
||||||
|
let intcode = read_intcode(TEST_INPUT1).unwrap();
|
||||||
|
let circuit = AmplificationCircuit::new(intcode, 5, &[4, 3, 2, 1, 0]);
|
||||||
|
assert_eq!(circuit.execute_circuit(0).unwrap(), 43210);
|
||||||
|
|
||||||
|
let intcode = read_intcode(TEST_INPUT2).unwrap();
|
||||||
|
let circuit = AmplificationCircuit::new(intcode, 5, &[0, 1, 2, 3, 4]);
|
||||||
|
assert_eq!(circuit.execute_circuit(0).unwrap(), 54321);
|
||||||
|
|
||||||
|
let intcode = read_intcode(TEST_INPUT3).unwrap();
|
||||||
|
let circuit = AmplificationCircuit::new(intcode, 5, &[1, 0, 4, 3, 2]);
|
||||||
|
assert_eq!(circuit.execute_circuit(0).unwrap(), 65210);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn finds_max_output_of_circuits() {
|
||||||
|
let inputs = [TEST_INPUT1, TEST_INPUT2, TEST_INPUT3];
|
||||||
|
let outputs = [43210, 54321, 65210];
|
||||||
|
for (input, output) in inputs.iter().zip(outputs.iter()) {
|
||||||
|
let intcode = read_intcode(input).unwrap();
|
||||||
|
let mut circuit = AmplificationCircuit::new(intcode, 5, &[0, 0, 0, 0, 0]);
|
||||||
|
assert_eq!(circuit.find_max_output(0).unwrap(), *output);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user